diff --git a/build_tools/rocm/run_xla.sh b/build_tools/rocm/run_xla.sh index bc2add44570ab..7fd68b6e35431 100755 --- 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 \