Add timeout limit to run_cpu.sh

ROCm · Sep 13, 2024 · e8bf71c · e8bf71c
1 parent ef54b72
commit e8bf71c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/tools/ci_build/linux/rocm/run_cpu.sh b/tensorflow/tools/ci_build/linux/rocm/run_cpu.sh
@@ -45,6 +45,7 @@ if [ -f /usertools/cpu.bazelrc ]; then
           --action_env=OPENBLAS_CORETYPE=Haswell \
           --action_env=TF_ENABLE_ONEDNN_OPTS=0 \
           --local_test_jobs=${N_BUILD_JOBS} \
+          --test_timeout 920,2400,7200,9600 \
           --jobs=${N_BUILD_JOBS}
 else
          yes "" | $PYTHON_BIN_PATH configure.py