From 88325db237e5940bbacc326b1c0395c295c41873 Mon Sep 17 00:00:00 2001
From: agentelement <agentelement@agentelement.net>
Date: Sat, 6 Sep 2025 15:24:53 -0700
Subject: [PATCH] rocmPackages: add gfx1151 to list of broken arches for
 pytorch

---
 .../python-modules/torch/source/default.nix            | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix
index ccd0ac5c6956..f5a082888bc8 100644
--- a/pkgs/development/python-modules/torch/source/default.nix
+++ b/pkgs/development/python-modules/torch/source/default.nix
@@ -175,12 +175,16 @@ let
     else if cudaSupport then
       gpuArchWarner supportedCudaCapabilities unsupportedCudaCapabilities
     else if rocmSupport then
-      # Remove RDNA1 gfx101x archs from default ROCm support list to avoid
-      # use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'
-      # TODO: Retest after ROCm 6.4 or torch 2.8
       lib.lists.subtractLists [
+        # Remove RDNA1 gfx101x archs from default ROCm support list to avoid
+        # use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'
+        # TODO: Retest after ROCm 6.4 or torch 2.8
         "gfx1010"
         "gfx1012"
+
+        # Strix Halo seems to be broken as well, see
+        # https://github.com/NixOS/nixpkgs/pull/440359.
+        "gfx1151"
       ] (rocmPackages.clr.localGpuTargets or rocmPackages.clr.gpuTargets)
     else
       throw "No GPU targets specified"