Skip to content

Commit d3cee1a

Browse files
committed
move cuda13 info to bazelrc
1 parent 2610b35 commit d3cee1a

File tree

1 file changed

+19
-7
lines changed

1 file changed

+19
-7
lines changed

deps/ReactantExtra/.bazelrc

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -32,21 +32,33 @@ build -c opt
3232
common:cuda --repo_env TF_NEED_CUDA=1
3333
common:cuda --repo_env TF_NVCC_CLANG=1
3434
common:cuda --repo_env TF_NCCL_USE_STUB=1
35-
common:cuda --repo_env=HERMETIC_CUDA_VERSION="12.8.1"
36-
common:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.8.0"
37-
common:cuda --repo_env=HERMETIC_NVSHMEM_VERSION="3.2.5"
38-
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
39-
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
40-
common:cuda --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_60,sm_70,sm_80,compute_90"
41-
common:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain"
4235
common:cuda --@local_config_cuda//:enable_cuda
36+
common:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain"
4337
# Default hermetic CUDA and CUDNN versions.
4438
common:cuda --@local_config_cuda//cuda:include_cuda_libs=true
4539
common:cuda --@local_config_cuda//:cuda_compiler=nvcc
4640
# common:cuda --@local_config_nvshmem//:override_include_nvshmem_libs=true
4741
# common:cuda --@local_config_nvshmem//cuda:include_nvshmem_libs=true
4842

4943

44+
common:cuda12 --config=cuda
45+
common:cuda12 --repo_env=HERMETIC_CUDA_VERSION="12.8.1"
46+
common:cuda12 --repo_env=HERMETIC_CUDNN_VERSION="9.8.0"
47+
common:cuda12 --repo_env=HERMETIC_NVSHMEM_VERSION="3.2.5"
48+
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
49+
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
50+
common:cuda12 --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_60,sm_70,sm_80,compute_90"
51+
52+
53+
common:cuda13 --config=cuda
54+
common:cuda13 --repo_env=HERMETIC_CUDA_VERSION="13.0.0"
55+
common:cuda13 --repo_env=HERMETIC_CUDNN_VERSION="9.12.0"
56+
common:cuda13 --repo_env=HERMETIC_NVSHMEM_VERSION="3.3.20"
57+
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
58+
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
59+
common:cuda13 --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_75,sm_80,sm_90,sm_100,compute_120"
60+
61+
5062
common:rocm --crosstool_top=@local_config_rocm//crosstool:toolchain
5163
common:rocm --define=using_rocm=true --define=using_rocm_hipcc=true
5264
common:rocm --repo_env TF_NEED_ROCM=1

0 commit comments

Comments
 (0)