python314Packages.torch: automatically adjust the magma cuda version

to improve overridability
2026-07-19 23:21:27 +00:00 · 2026-07-08 02:08:12 +02:00
parent f71a0a88b3
commit 6150d2f95a
1 changed files with 1 additions and 1 deletions
--- a/pkgs/development/python-modules/torch/source/default.nix
+++ b/pkgs/development/python-modules/torch/source/default.nix
@@ -14,7 +14,7 @@
  autoAddDriverRunpath,
  effectiveMagma ?
    if cudaSupport then
-      magma-cuda-static
+      magma-cuda-static.override { inherit cudaPackages; }
    else if rocmSupport then
      magma-hip
    else