Add flaky_test_attempts to run_xla.sh

ROCm · Jun 26, 2024 · adf7949 · adf7949
1 parent 2471a91
commit adf7949
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/build_tools/rocm/run_xla.sh b/build_tools/rocm/run_xla.sh
@@ -60,6 +60,7 @@ bazel \
     --test_sharding_strategy=disabled \
     --test_output=errors \
     --keep_going \
+    --flaky_test_attempts=3 \
     --local_test_jobs=${N_TEST_JOBS} \
     --test_env=TF_TESTS_PER_GPU=$TF_TESTS_PER_GPU \
     --test_env=TF_GPU_COUNT=$TF_GPU_COUNT \