Skip to content

Commit f8bbcb8

Browse files
committed
nvcc clang
1 parent 44e50f4 commit f8bbcb8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

deps/ReactantExtra/.bazelrc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,20 +15,20 @@ build --define=allow_oversize_protos=true
1515
build -c opt
1616

1717
build:cuda --repo_env TF_NEED_CUDA=1
18+
build:cuda --repo_env TF_NVCC_CLANG=1
1819
build:cuda --repo_env TF_NCCL_USE_STUB=1
1920
build:cuda --repo_env=HERMETIC_CUDA_VERSION="12.3.2"
2021
build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.1.1"
2122
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
2223
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
2324
build:cuda --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,compute_90"
24-
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
25+
build:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain"
2526
build:cuda --@local_config_cuda//:enable_cuda
2627
build:cuda --@xla//xla/python:jax_cuda_pip_rpaths=true
2728
# Default hermetic CUDA and CUDNN versions.
2829
build:cuda --@local_config_cuda//cuda:include_cuda_libs=true
2930
build:cuda --@local_config_cuda//:cuda_compiler=nvcc
3031

31-
3232
build:rocm --repo_env TF_NEED_ROCM=1
3333
build:rocm --define=using_rocm=true
3434
build:rocm --define=using_rocm_hipcc=true

0 commit comments

Comments
 (0)