feat(rocke): add gfx950 scaled MFMA GEMM examples, 32x32x64 tile

dnn-providers/hip-kernel-provider/rocke/platform/cpp/bindings/rocke_engine.cpp

@@ -52,6 +52,7 @@ extern "C" {
 #include "rocke/instance_gemm_multi_abd.h"
 #include "rocke/instance_gemm_multi_d.h"
 #include "rocke/instance_gemm_universal.h"
+#include "rocke/instance_gfx950_block_scaled_gemm.h"
 #include "rocke/instance_grouped_gemm.h"
 #include "rocke/instance_matmul_nbits.h"
 #include "rocke/instance_mfma_gemm.h"
@@ -1031,6 +1032,63 @@ std::vector<std::string> mx_gemm_verify(const py::dict& d, const std::string& ar
                              rocke_build_mx_gemm_new(&b, &s, arch_or_default(arch)));
 }
 
+rocke_gfx950_block_scaled_gemm_spec_t gfx950_scaled_build_spec(const py::dict& d,
+                                                               std::deque<std::string>& store)
+{
+    auto s = rocke_gfx950_block_scaled_gemm_spec_default();
+    s.M = dict_int(d, "M", s.M);
+    s.N = dict_int(d, "N", s.N);
+    s.K = dict_int(d, "K", s.K);
+    s.block_k = dict_int(d, "block_k", s.block_k);
+    const char* fields[] = {"name", "dtype_a", "dtype_b", "dtype_c"};
+    const char** outputs[] = {&s.name, &s.dtype_a, &s.dtype_b, &s.dtype_c};
+    for(int i = 0; i < 4; ++i)
+    {
+        std::string v;
+        if(dict_str(d, fields[i], v))
+        {
+            store.push_back(v);
+            *outputs[i] = store.back().c_str();
+        }
+    }
+    return s;
+}
+std::string gfx950_scaled_serialize_ir(const py::dict& d, const std::string& arch)
+{
+    ROCKE_FAMILY_SERIALIZE_BODY(
+        "rocke_engine.gfx950_block_scaled_gemm_serialize_ir",
+        rocke_gfx950_block_scaled_gemm_spec_t,
+        gfx950_scaled_build_spec,
+        rocke_build_gfx950_block_scaled_gemm_new(&b, &s, arch_or_default(arch)));
+}
+std::vector<std::string> gfx950_scaled_verify(const py::dict& d, const std::string& arch)
+{
+    ROCKE_FAMILY_VERIFY_BODY(
+        "rocke_engine.gfx950_block_scaled_gemm_verify",
+        rocke_gfx950_block_scaled_gemm_spec_t,
+        gfx950_scaled_build_spec,
+        rocke_build_gfx950_block_scaled_gemm_new(&b, &s, arch_or_default(arch)));
+}
+std::string gfx950_scaled_lower_llvm(const py::dict& d, const std::string& arch)
+{
+    std::deque<std::string> store;
+    auto s = gfx950_scaled_build_spec(d, store);
+    rocke_ir_builder_t b;
+    auto* k = rocke_build_gfx950_block_scaled_gemm_new(&b, &s, arch_or_default(arch));
+    if(!k)
+    {
+        std::string err = rocke_ir_builder_error(&b);
+        rocke_ir_builder_free(&b);
+        throw std::runtime_error(err);
+    }
+    char* ll = nullptr;
+    char err[ROCKE_ERR_MSG_CAP] = {};
+    auto status = rocke_lower_kernel_to_llvm_ex(
+        k, ROCKE_LLVM_FLAVOR_AUTO, arch_or_default(arch), &ll, err, sizeof(err));
+    rocke_ir_builder_free(&b);
+    return take_lowered(status, ll, err, "gfx950_block_scaled_gemm");
+}
+
 /* ============================== mfma GEMM ============================= */
 
 rocke_mfma_gemm_spec_t mfma_build_spec(const py::dict& d, std::deque<std::string>& store)
@@ -3787,6 +3845,10 @@ PYBIND11_MODULE(rocke_engine, m)
          &block_scale_gemm_lower_llvm,
          &block_scale_gemm_serialize_ir,
          &block_scale_gemm_verify);
+    reg3("gfx950_block_scaled_gemm",
+         &gfx950_scaled_lower_llvm,
+         &gfx950_scaled_serialize_ir,
+         &gfx950_scaled_verify);
     reg3("mx_gemm", &mx_gemm_lower_llvm, &mx_gemm_serialize_ir, &mx_gemm_verify);
     reg3("mfma_gemm", &mfma_gemm_lower_llvm, &mfma_gemm_serialize_ir, &mfma_gemm_verify);
     m.def("mfma_gemm_is_valid", &mfma_gemm_is_valid, py::arg("spec"), py::arg("arch") = "gfx950");