fix: hipBLASLt support for gfx15X has been added starting from ROCm 6.4.2

Author: slojosic-amd

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

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"