Skip to content

Commit ee15a6b

Browse files
committed
Unset TF_NEED_ROCM for cpu tests
1 parent 1231900 commit ee15a6b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ bazel --bazelrc=tensorflow/tools/tf_sig_build_dockerfiles/devel.usertools/cpu.ba
3737
--config=sigbuild_local_cache \
3838
--verbose_failures \
3939
--config=pycpp \
40-
--action_env=TF_NEED_ROCM=1 \
40+
--action_env=TF_NEED_ROCM=0 \
4141
--action_env=TF_PYTHON_VERSION=$PYTHON_VERSION \
4242
--local_test_jobs=${N_BUILD_JOBS} \
43-
--repo_env=ROCM_PATH=/opt/rocm \
43+
--repo_env=ROCM_PATH=/opt/rocm \
4444
--jobs=${N_BUILD_JOBS}

0 commit comments

Comments
 (0)