Skip to content

Commit 4d8cc48

Browse files
committed
[r2.20] Use timeout from bazelrc and add ROCM_PATH repo_env
1 parent 879bc1c commit 4d8cc48

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -59,25 +59,26 @@ if [ -z "$TARGET_ARCHS" ]; then
5959
exit 1
6060
fi
6161

62+
if [ ! -d /tf ];then
63+
# The bazelrc files in /usertools expect /tf to exist
64+
mkdir /tf
65+
fi
66+
6267
# Run bazel test command. Double test timeouts to avoid flakes.
63-
bazel test \
68+
bazel --bazelrc=tensorflow/tools/tf_sig_build_dockerfiles/devel.usertools/rocm.bazelrc test \
6469
--config=rocm \
70+
--config=sigbuild_local_cache \
71+
--config=pycpp \
6572
-k \
66-
--test_tag_filters=gpu,-no_oss,-oss_excluded,-oss_serial,-no_gpu,-cuda-only,-benchmark-test,-rocm_multi_gpu,-tpu,-v1only \
6773
--jobs=${N_BUILD_JOBS} \
6874
--local_test_jobs=${N_TEST_JOBS} \
6975
--test_env=TF_GPU_COUNT=$TF_GPU_COUNT \
7076
--test_env=TF_TESTS_PER_GPU=$TF_TESTS_PER_GPU \
7177
--test_env=MIOPEN_DEBUG_CONV_WINOGRAD=0 \
72-
--test_timeout 600,900,2400,7200 \
7378
--repo_env="TF_ROCM_AMDGPU_TARGETS=$TARGET_ARCHS" \
79+
--repo_env="ROCM_PATH=/opt/rocm" \
7480
--build_tests_only \
7581
--test_output=errors \
82+
--verbose_failures \
7683
--test_sharding_strategy=disabled \
77-
--test_size_filters=small,medium,large \
78-
--run_under=//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute \
79-
-- \
80-
//tensorflow/... \
81-
-//tensorflow/core/tpu/... \
82-
-//tensorflow/lite/... \
83-
-//tensorflow/compiler/tf2tensorrt/... \
84+
--run_under=//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute

0 commit comments

Comments
 (0)