python3Packages.torchWithRocm: mark as broken on non-default python versions

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" ''