Merge pull request #3561 from ROCm/gfx12_bp

Backport gfx12 improvements to 6.3 release
ROCm · Nov 1, 2024 · 49ede4d · 49ede4d
2 parents 0fb79b4 + f85c896
commit 49ede4d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/targets/gpu/fuse_mlir.cpp b/src/targets/gpu/fuse_mlir.cpp
@@ -983,7 +983,7 @@ void fuse_mlir::apply(module_pass_manager& mpm) const
 #ifdef MIGRAPHX_MLIR
     std::size_t counter     = 0;
     const auto& device_name = ctx == nullptr ? "" : ctx->get_current_device().get_gfx_name();
-    const bool is_navi      = starts_with(device_name, "gfx11");
+    const bool is_navi = starts_with(device_name, "gfx11") or starts_with(device_name, "gfx12");
 
     auto get_mode = [&](std::string_view option, mlir_mode m1, mlir_mode m2 = mlir_mode::fast) {
         if(specific_op<rejected>(option))

diff --git a/src/targets/gpu/hipblaslt.cpp b/src/targets/gpu/hipblaslt.cpp
@@ -56,9 +56,10 @@ hipblaslt_preference_ptr create_hipblaslt_preference_ptr()
 bool hipblaslt_supported()
 {
     const auto device_name = trim(split_string(get_device_name(), ':').front());
-    // hipblaslt is supported for MI100 and above and Navi3x and above
-    return (starts_with(device_name, "gfx9") and device_name >= "gfx908" and
-            not starts_with(device_name, "gfx10"));
+    // hipblaslt is supported for MI200 and above, and Navi3x and above.
+    return (device_name == "gfx90a" or
+            (starts_with(device_name, "gfx94") and device_name >= "gfx940") or
+            starts_with(device_name, "gfx110") or starts_with(device_name, "gfx120"));
 }
 
 #endif // MIGRAPHX_USE_HIPBLASLT