From e8b457bc1c12b560740bb8cb664b657966ccd18b Mon Sep 17 00:00:00 2001
From: Antoine Viallon <antoine@lesviallon.fr>
Date: Wed, 26 Mar 2025 00:09:43 +0100
Subject: [PATCH] [Hardware/AMD/ROCm] override rocmPackages.clr with custom
 supported architectures

---
 hardware/amd/rocm.nix | 34 ++++++++++++++++++++++++++++++----
 1 file changed, 30 insertions(+), 4 deletions(-)

diff --git a/hardware/amd/rocm.nix b/hardware/amd/rocm.nix
index 1908603..11a78f7 100644
--- a/hardware/amd/rocm.nix
+++ b/hardware/amd/rocm.nix
@@ -47,6 +47,26 @@ in {
       example = "gfx902";
       type = types.string;
     };
+    gpuTargets = mkOption {
+      description = "Override supported GPU ISAs in some ROCm packages.";
+      default = [ "803"
+                "900"
+                "906:xnack-"
+                "908:xnack-"
+                "90a:xnack+" "90a:xnack-"
+                "940"
+                "941"
+                "942"
+                "1010"
+                "1012"
+                "1030"
+                "1031"
+                "1100"
+                "1101"
+                "1102" ];
+      example = [ "900" "1031" ];
+      type = with types; nullOr (listOf string);
+    };
   };
 
   config = mkIf (cfg.enable && localCfg.enable) {  
@@ -86,9 +106,15 @@ in {
     nixpkgs.config.rocmSupport = true;
 
     nixpkgs.overlays = mkBefore [(final: prev: {
-        # Overlay Blender to use the HIP build if we have a compatible AMD GPU
-        blender = prev.blender-hip;
-        blender-cpu = prev.blender;
-      })];
+        rocmPackages = prev.rocmPackages // {
+          clr = prev.rocmPackages.clr.overrideAttrs (oldAttrs: {
+            passthru = oldAttrs.passthru // {
+              # We cannot use this for each ROCm library, as each defines their own supported targets
+              # See: https://github.com/ROCm/ROCm/blob/77cbac4abab13046ee93d8b5bf410684caf91145/README.md#library-target-matrix
+              gpuTargets = lib.forEach localCfg.gpuTargets (target: "gfx${target}");
+            };
+          });
+        };
+    })];
   };
 }