File tree 1 file changed +3
-2
lines changed
1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ build:native_arch_posix --host_copt=-march=native
47
47
build:mkl_open_source_only --define=tensorflow_mkldnn_contraction_kernel=1
48
48
49
49
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
51
51
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
52
52
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
53
53
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
@@ -92,7 +92,8 @@ build:tpu --define=with_tpu_support=true
92
92
test:tpu --local_test_jobs=1
93
93
test:cuda --local_test_jobs=1
94
94
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
96
97
97
98
#########################################################################
98
99
# RBE config options below.
You can’t perform that action at this time.
0 commit comments