Skip to content

Commit e5f8f74

Browse files
committed
Enable TF_NEED_ROCM in cpu tests script
1 parent b6bd221 commit e5f8f74

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ 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=0 \
40+
--action_env=TF_NEED_ROCM=1 \
4141
--action_env=TF_PYTHON_VERSION=$PYTHON_VERSION \
4242
--local_test_jobs=${N_BUILD_JOBS} \
4343
--repo_env=ROCM_PATH=/opt/rocm \

0 commit comments

Comments
 (0)