Skip to content

Commit 158b986

Browse files
committed
fix
1 parent 1737a70 commit 158b986

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

deps/ReactantExtra/.bazelrc

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,8 @@ common:macos --define ynn_enable_arm64_sme=false
3434
common:cuda --repo_env TF_NEED_CUDA=1
3535
common:cuda --repo_env TF_NVCC_CLANG=1
3636
common: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
3839
common:cuda --@local_config_cuda//:enable_cuda
3940
common: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

4647
common:cuda12 --config=cuda
48+
common:cuda12 --config=cuda_static
4749
common: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"
4951
common: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.
5254
common:cuda12 --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,sm_90,sm_100,compute_120"
5355

5456

5557
common: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"
5860
common: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.

deps/ReactantExtra/WORKSPACE

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ NSYNC_COMMIT = "82b118aa7ace3132e517e2c467f8732978cf4023"
44

55
NSYNC_SHA256 = ""
66

7-
ENZYMEXLA_COMMIT = "c4ad72b48c57c448e7fe6443e1d7eb94256be40b"
7+
ENZYMEXLA_COMMIT = "8dc549db2c67dd4940743b6fbca96af2cab41de5"
88

99
ENZYMEXLA_SHA256 = ""
1010

0 commit comments

Comments
 (0)