Skip to content

Commit 321bb36

Browse files
committed
NCCL fix.
1 parent 148c798 commit 321bb36

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

.bazelrc

+3-2
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ build:native_arch_posix --host_copt=-march=native
4747
build:mkl_open_source_only --define=tensorflow_mkldnn_contraction_kernel=1
4848

4949
build:cuda --repo_env TF_NEED_CUDA=1
50-
build:cuda --repo_env TF_NCCL_USE_STUB=1
50+
build:cuda --repo_env TF_NCCL_USE_STUB=0
5151
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
5252
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
5353
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
@@ -92,7 +92,8 @@ build:tpu --define=with_tpu_support=true
9292
test:tpu --local_test_jobs=1
9393
test:cuda --local_test_jobs=1
9494

95-
test:cuda --@local_config_cuda//cuda:include_cuda_libs=true
95+
test:cuda --repo_env TF_NCCL_USE_STUB=1
96+
test:cuda --@local_config_cuda//cuda:include_cuda_libs=false
9697

9798
#########################################################################
9899
# RBE config options below.

0 commit comments

Comments
 (0)