From b8bf96307449423ce014ec608fc1a0a1348f8033 Mon Sep 17 00:00:00 2001
From: Milica Stankovic <mstankov@amd.com>
Date: Tue, 26 Aug 2025 18:27:55 +0200
Subject: [PATCH 1/2] Add gfx1150/gfx1151 to hipBLASLt supported architecture
 list

---
 aten/src/ATen/Context.cpp          | 2 +-
 aten/src/ATen/native/cuda/Blas.cpp | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp
index fefcf731cd825..1a679078281e2 100644
--- a/aten/src/ATen/Context.cpp
+++ b/aten/src/ATen/Context.cpp
@@ -336,7 +336,7 @@ at::BlasBackend Context::blasPreferredBackend() {
       static const std::vector<std::string> archs = {
           "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60400
-          "gfx1200", "gfx1201",
+          "gfx1150", "gfx1151", "gfx1200", "gfx1201",
 #endif
 #if ROCM_VERSION >= 60500
           "gfx950"
diff --git a/aten/src/ATen/native/cuda/Blas.cpp b/aten/src/ATen/native/cuda/Blas.cpp
index 1fc9e14189e46..2fc9cdddaf66d 100644
--- a/aten/src/ATen/native/cuda/Blas.cpp
+++ b/aten/src/ATen/native/cuda/Blas.cpp
@@ -270,7 +270,7 @@ static bool isSupportedHipLtROCmArch(int index) {
     static const std::vector<std::string> archs = {
         "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60300
-        "gfx1100", "gfx1101", "gfx1200", "gfx1201",
+        "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1200", "gfx1201",
 #endif
 #if ROCM_VERSION >= 60500
         "gfx950"

From 39aaf5309b75b09e64b6a6189ed2b2026ac99fab Mon Sep 17 00:00:00 2001
From: Slobodan Josic <Slobodan.Josic@amd.com>
Date: Fri, 24 Oct 2025 11:19:44 +0200
Subject: [PATCH 2/2] fix: hipBLASLt support for gfx15X has been added starting
 from ROCm 6.4.2

---
 aten/src/ATen/Context.cpp          | 8 +++++++-
 aten/src/ATen/native/cuda/Blas.cpp | 5 ++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp
index 1a679078281e2..a1d8b696c50a2 100644
--- a/aten/src/ATen/Context.cpp
+++ b/aten/src/ATen/Context.cpp
@@ -336,7 +336,10 @@ at::BlasBackend Context::blasPreferredBackend() {
       static const std::vector<std::string> archs = {
           "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60400
-          "gfx1150", "gfx1151", "gfx1200", "gfx1201",
+          "gfx1200", "gfx1201",
+#endif
+#if ROCM_VERSION >= 60402
+          "gfx1150", "gfx1151",
 #endif
 #if ROCM_VERSION >= 60500
           "gfx950"
@@ -364,6 +367,9 @@ at::BlasBackend Context::blasPreferredBackend() {
 #if ROCM_VERSION >= 60300
           "gfx1100", "gfx1101", "gfx1200", "gfx1201",
 #endif
+#if ROCM_VERSION >= 60402
+          "gfx1150", "gfx1151",
+#endif
 #if ROCM_VERSION >= 60500
           "gfx950"
 #endif
diff --git a/aten/src/ATen/native/cuda/Blas.cpp b/aten/src/ATen/native/cuda/Blas.cpp
index 2fc9cdddaf66d..aea89c3d4eb24 100644
--- a/aten/src/ATen/native/cuda/Blas.cpp
+++ b/aten/src/ATen/native/cuda/Blas.cpp
@@ -270,7 +270,10 @@ static bool isSupportedHipLtROCmArch(int index) {
     static const std::vector<std::string> archs = {
         "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60300
-        "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1200", "gfx1201",
+        "gfx1100", "gfx1101", "gfx1200", "gfx1201",
+#endif
+#if ROCM_VERSION >= 60402
+        "gfx1150", "gfx1151",
 #endif
 #if ROCM_VERSION >= 60500
         "gfx950"