From d22db82c1ee491f2b436fce3f464933129eae70a Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 30 Mar 2026 21:37:48 +0000
Subject: [PATCH] python3Packages.torch-bin: fix tests evaluation

---
 pkgs/development/python-modules/torch/bin/default.nix | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pkgs/development/python-modules/torch/bin/default.nix b/pkgs/development/python-modules/torch/bin/default.nix
index 3e6ae61234aa..9f2e6fb7e7c9 100644
--- a/pkgs/development/python-modules/torch/bin/default.nix
+++ b/pkgs/development/python-modules/torch/bin/default.nix
@@ -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";