Skip to content

Commit 5751b52

Browse files
committed
Static cuda attempt
1 parent 96a819d commit 5751b52

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

deps/ReactantExtra/.bazelrc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ 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
3738
common:cuda --@local_config_cuda//:enable_cuda
3839
common:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain"
3940
# Default hermetic CUDA and CUDNN versions.

deps/ReactantExtra/WORKSPACE

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ LLVM_TARGETS = [
7979

8080
load("@enzyme_ad//third_party/ml_toolchain:workspace.bzl", ml_toolchain_workspace = "repo")
8181

82-
ml_toolchain_workspace()
82+
ml_toolchain_workspace("6bc326eeb70f8b3d28a06ec719e0c40fd909b07a")
8383

8484
load("@enzyme_ad//third_party/jax:workspace.bzl", jax_workspace = "repo")
8585

0 commit comments

Comments
 (0)