bug fix

2026-07-11 01:27:34 +00:00 · 2025-12-22 01:51:50 +00:00
parent c08987af05
commit 56aa5385c6
1 changed files with 8 additions and 1 deletions
--- a/example/ck_tile/17_grouped_gemm/grouped_gemm.cpp
+++ b/example/ck_tile/17_grouped_gemm/grouped_gemm.cpp
@@ -271,7 +271,14 @@ int run_gemm_example_prec_type(std::string a_layout, std::string b_layout, std::
                                                     CDataType,
                                                     AccDataType>(argc, argv, Row{}, Col{}, Col{});
    }
-
+    else if(a_layout == "C" && b_layout == "R" && c_layout == "C")
+    {
+        return run_grouped_gemm_example_with_layouts<GemmConfig,
+                                                     ADataType,
+                                                     BDataType,
+                                                     CDataType,
+                                                     AccDataType>(argc, argv, Col{}, Row{}, Col{});
+    }
    else
    {
        throw std::runtime_error("Unsupported data layout configuration for A, B and C tensors!");