@@ -34,7 +34,8 @@ common:macos --define ynn_enable_arm64_sme=false
3434common:cuda --repo_env TF_NEED_CUDA=1
3535common:cuda --repo_env TF_NVCC_CLANG=1
3636common:cuda --repo_env TF_NCCL_USE_STUB=1
37- common:cuda --@rules_ml_toolchain//common:link_cuda_static_libs=true
37+ common:cuda_static --@rules_ml_toolchain//common:link_cuda_static_libs=true
38+ common:cuda_static --@rules_ml_toolchain//common:link_nvrtc_static_libs=true
3839common:cuda --@local_config_cuda//:enable_cuda
3940common:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain"
4041# Default hermetic CUDA and CUDNN versions.
@@ -44,17 +45,18 @@ common:cuda --@local_config_cuda//:cuda_compiler=nvcc
4445# common:cuda --@local_config_nvshmem//cuda:include_nvshmem_libs=true
4546
4647common:cuda12 --config=cuda
48+ common:cuda12 --config=cuda_static
4749common:cuda12 --repo_env=HERMETIC_CUDA_VERSION="12.9.1"
48- common:cuda12 --repo_env=HERMETIC_CUDNN_VERSION="9.13 .0"
50+ common:cuda12 --repo_env=HERMETIC_CUDNN_VERSION="9.14 .0"
4951common:cuda12 --repo_env=HERMETIC_NVSHMEM_VERSION="3.3.9"
5052# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
5153# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
5254common:cuda12 --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,sm_90,sm_100,compute_120"
5355
5456
5557common:cuda13 --config=cuda
56- common:cuda13 --repo_env=HERMETIC_CUDA_VERSION="13.0.0 "
57- common:cuda13 --repo_env=HERMETIC_CUDNN_VERSION="9.12 .0"
58+ common:cuda13 --repo_env=HERMETIC_CUDA_VERSION="13.0.2 "
59+ common:cuda13 --repo_env=HERMETIC_CUDNN_VERSION="9.14 .0"
5860common:cuda13 --repo_env=HERMETIC_NVSHMEM_VERSION="3.3.20"
5961# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
6062# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
0 commit comments