From 7ad3e7f8086ac5575d14d7a259192060c3dce39c Mon Sep 17 00:00:00 2001
From: Filip Jankovic <161825881+fjankovi@users.noreply.github.com>
Date: Wed, 14 May 2025 21:05:59 +0200
Subject: [PATCH] [release/2.7] Prefer hipblaslt for gfx1200, gfx1201 (#2125)

---
 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 300fdb36bb90e..3113a6784d3fe 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