From 96d08441542c9f74c5c8bcb001df3a532489c2ba Mon Sep 17 00:00:00 2001
From: Vuksan Mijovic <vmijovic@amd.com>
Date: Thu, 23 Oct 2025 18:48:21 +0200
Subject: [PATCH 1/2] update support/prefer hipblaslt architectures

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

diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp
index 7a8d02be530e3..4e39cdeacaa35 100644
--- a/aten/src/ATen/Context.cpp
+++ b/aten/src/ATen/Context.cpp
@@ -472,7 +472,7 @@ at::BlasBackend Context::blasPreferredBackend() {
       static const std::vector<std::string> archs = {
           "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60400
-          "gfx1200", "gfx1201",
+          "gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
 #endif
 #if ROCM_VERSION >= 60500
           "gfx950"
@@ -502,7 +502,7 @@ at::BlasBackend Context::blasPreferredBackend() {
       static const std::vector<std::string> archs = {
           "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60300
-          "gfx1100", "gfx1101", "gfx1200", "gfx1201", "gfx908",
+          "gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
 #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 23447c7e09b3f..d0301ee51dabd 100644
--- a/aten/src/ATen/native/cuda/Blas.cpp
+++ b/aten/src/ATen/native/cuda/Blas.cpp
@@ -283,7 +283,7 @@ static bool isSupportedHipLtROCmArch(int index) {
     static const std::vector<std::string> archs = {
         "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60300
-        "gfx1100", "gfx1101", "gfx1200", "gfx1201", "gfx908",
+        "gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
 #endif
 #if ROCM_VERSION >= 60500
         "gfx950"

From 151b746dcd92a12c80c8a5dd80cb503aae2d15f8 Mon Sep 17 00:00:00 2001
From: Vuksan Mijovic <vmijovic@amd.com>
Date: Fri, 24 Oct 2025 14:10:53 +0200
Subject: [PATCH 2/2] add gfx1151 and gfx1102

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

diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp
index 4e39cdeacaa35..eafaff8d0dcc2 100644
--- a/aten/src/ATen/Context.cpp
+++ b/aten/src/ATen/Context.cpp
@@ -471,8 +471,11 @@ at::BlasBackend Context::blasPreferredBackend() {
     static const bool hipblaslt_preferred = []() {
       static const std::vector<std::string> archs = {
           "gfx90a", "gfx942",
-#if ROCM_VERSION >= 60400
-          "gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
+#if ROCM_VERSION >= 60300
+          "gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",
+#endif
+#if ROCM_VERSION >= 60402
+          "gfx1150", "gfx1151",
 #endif
 #if ROCM_VERSION >= 60500
           "gfx950"
@@ -502,7 +505,10 @@ at::BlasBackend Context::blasPreferredBackend() {
       static const std::vector<std::string> archs = {
           "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60300
-          "gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
+          "gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", "gfx908",
+#endif
+#if ROCM_VERSION >= 60402
+          "gfx1150", "gfx1151",
 #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 d0301ee51dabd..49356f8c79bc8 100644
--- a/aten/src/ATen/native/cuda/Blas.cpp
+++ b/aten/src/ATen/native/cuda/Blas.cpp
@@ -283,7 +283,10 @@ static bool isSupportedHipLtROCmArch(int index) {
     static const std::vector<std::string> archs = {
         "gfx90a", "gfx942",
 #if ROCM_VERSION >= 60300
-        "gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
+        "gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", "gfx908",
+#endif
+#if ROCM_VERSION >= 60402
+          "gfx1150", "gfx1151",
 #endif
 #if ROCM_VERSION >= 60500
         "gfx950"