File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -34,6 +34,7 @@ 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
3738common:cuda --@local_config_cuda//:enable_cuda
3839common:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain"
3940# Default hermetic CUDA and CUDNN versions.
Original file line number Diff line number Diff line change @@ -79,7 +79,7 @@ LLVM_TARGETS = [
7979
8080load ("@enzyme_ad//third_party/ml_toolchain:workspace.bzl" , ml_toolchain_workspace = "repo" )
8181
82- ml_toolchain_workspace ()
82+ ml_toolchain_workspace ("6bc326eeb70f8b3d28a06ec719e0c40fd909b07a" )
8383
8484load ("@enzyme_ad//third_party/jax:workspace.bzl" , jax_workspace = "repo" )
8585
You can’t perform that action at this time.
0 commit comments