Merge pull request #2334 from ROCmSoftwarePlatform/fix-timeout

Fix timeout and run_under
ROCm · Dec 19, 2023 · 795c848 · 795c848
2 parents 04c9ce6 + 7c70659
commit 795c848
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/devel.usertools/gpu.bazelrc b/tensorflow/tools/tf_sig_build_dockerfiles/devel.usertools/gpu.bazelrc
@@ -20,6 +20,7 @@ build --action_env=CACHEBUSTER=565341047
 
 # Test-related settings below this point.
 test:rocm --test_env=HSA_TOOLS_LIB=libroctracer64.so --test_sharding_strategy=disabled
+test --test_timeout=920,2400,7200,9600 --local_test_jobs=4 --run_under=//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute
 
 # "nonpip" tests are regular py_test tests.
 # Pass --config=nonpip to run the same suite of tests. If you want to run just