From ab2d7e40168939df18c44a68f3d59098a24ccb56 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sat, 30 Aug 2025 00:47:36 +0200
Subject: [PATCH] python3Packages.torchWithRocm: mark as broken on non-default
 python versions

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

diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix
index 56396da3f78d..33ada3baab2c 100644
--- a/pkgs/development/python-modules/torch/source/default.nix
+++ b/pkgs/development/python-modules/torch/source/default.nix
@@ -1,6 +1,7 @@
 {
   stdenv,
   lib,
+  pkgs,
   fetchFromGitHub,
   fetchFromGitLab,
   fetchpatch2,
@@ -250,6 +251,9 @@ let
     "Magma cudaPackages does not match cudaPackages" =
       cudaSupport
       && (effectiveMagma.cudaPackages.cudaMajorMinorVersion != cudaPackages.cudaMajorMinorVersion);
+    # Should be fixed by WIP https://github.com/NixOS/nixpkgs/pull/438399
+    "ROCm support on non-default python versions is temporarily broken" =
+      rocmSupport && (pkgs.python3.version != python.version);
   };
 
   unroll-src = writeShellScript "unroll-src" ''