From 68a8346d785026f4dca4d5aa51e73433fdeeee2e Mon Sep 17 00:00:00 2001
From: Filip Jankovic <filip.jankovic@amd.com>
Date: Tue, 13 May 2025 12:00:13 +0200
Subject: [PATCH] Prefer hipblaslt fot gfx1200, gfx1201

---
 aten/src/ATen/Context.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp
index 01f223f4e5ce7..43875d938e1f5 100644
--- a/aten/src/ATen/Context.cpp
+++ b/aten/src/ATen/Context.cpp
@@ -335,6 +335,9 @@ at::BlasBackend Context::blasPreferredBackend() {
     static const bool hipblaslt_preferred = []() {
       static const std::vector<std::string> archs = {
           "gfx90a", "gfx942",
+#if ROCM_VERSION >= 60300
+          "gfx1200", "gfx1201",
+#endif
 #if ROCM_VERSION >= 60500
           "gfx950"
 #endif