[ROCm] Prefer hipblaslt for gfx1200, gfx1201 (pytorch#153610)

Pull Request resolved: pytorch#153610
Approved by: https://github.com/jeffdaily, https://github.com/atalman

Author: fjankovi

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 >= 60400
+          "gfx1200", "gfx1201",
+#endif
 #if ROCM_VERSION >= 60500
           "gfx950"
 #endif