diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py index e8829fc4a1..ead7bd4308 100755 --- a/tools/ci_build/build.py +++ b/tools/ci_build/build.py @@ -704,7 +704,9 @@ def main(): onnx_test_data_dir = os.path.join(source_dir, "cmake", "external", "onnx", "onnx", "backend", "test", "data", "simple") run_onnx_tests(build_dir, configs, onnx_test_data_dir, 'tensorrt', False, 1) elif args.use_cuda: - run_onnx_tests(build_dir, configs, onnx_test_data_dir, 'cuda', False, 2) + # Disable the test for flaky behavior on cuda9. Will remove the test case later + # run_onnx_tests(build_dir, configs, onnx_test_data_dir, 'cuda', False, 2) + pass elif args.x86 or platform.system() == 'Darwin': run_onnx_tests(build_dir, configs, onnx_test_data_dir, None, True, 1) # TODO: parallel executor test fails on MacOS