From 5a8f87d54c667b3e7acaee197bbf4279abe4b724 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= <me@danieldk.eu>
Date: Tue, 22 Sep 2026 13:47:36 +0000
Subject: [PATCH] kernel-builder: add `gfx11{02,03,50,51}` ROCm archs

Fixes #834.
---
 examples/kernels/flake.nix                                  | 4 ++++
 kernel-builder/src/pyproject/templates/torch/preamble.cmake | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/examples/kernels/flake.nix b/examples/kernels/flake.nix
index 325c44fe..553ad75c 100644
--- a/examples/kernels/flake.nix
+++ b/examples/kernels/flake.nix
@@ -242,6 +242,10 @@
             "gfx1030"
             "gfx1100"
             "gfx1101"
+            "gfx1102"
+            "gfx1103"
+            "gfx1150"
+            "gfx1151"
             "gfx1200"
             "gfx1201"
           ];
diff --git a/kernel-builder/src/pyproject/templates/torch/preamble.cmake b/kernel-builder/src/pyproject/templates/torch/preamble.cmake
index b02cbe8a..a3ce7cfb 100644
--- a/kernel-builder/src/pyproject/templates/torch/preamble.cmake
+++ b/kernel-builder/src/pyproject/templates/torch/preamble.cmake
@@ -22,7 +22,7 @@ include(FetchContent)
 file(MAKE_DIRECTORY ${FETCHCONTENT_BASE_DIR}) # Ensure the directory exists
 message(STATUS "FetchContent base directory: ${FETCHCONTENT_BASE_DIR}")
 
-set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1200;gfx1201")
+set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1200;gfx1201")
 
 # Make Torch CMake machinery happy. Whatever we use does not matter, since
 # we set the arches per-file later anyway.