python3Packages.torch-bin: fix tests evaluation

2026-03-30 21:37:48 +00:00
parent 7919000b68
commit d22db82c1e
1 changed files with 4 additions and 1 deletions
@@ -27,6 +27,7 @@
  typing-extensions,
  triton,

+  config,
  callPackage,
 }:

@@ -135,7 +136,9 @@ buildPythonPackage {

  pythonImportsCheck = [ "torch" ];

-  passthru.tests = callPackage ../tests { };
+  passthru.tests = callPackage ../tests {
+    inherit (config) rocmSupport cudaSupport;
+  };

  meta = {
    description = "PyTorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration";