@@ -32,21 +32,33 @@ build -c opt
3232common:cuda --repo_env TF_NEED_CUDA=1
3333common:cuda --repo_env TF_NVCC_CLANG=1
3434common: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"
4235common:cuda --@local_config_cuda//:enable_cuda
36+ common:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain"
4337# Default hermetic CUDA and CUDNN versions.
4438common:cuda --@local_config_cuda//cuda:include_cuda_libs=true
4539common: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+
5062common:rocm --crosstool_top=@local_config_rocm//crosstool:toolchain
5163common:rocm --define=using_rocm=true --define=using_rocm_hipcc=true
5264common:rocm --repo_env TF_NEED_ROCM=1
0 commit comments