[CK TILE GEMM] set correct value to TiledMMAPermuteN_ (#2839)

- TiledMMAPermuteN_ should be set to true when config if GemmConfigPreshufflePrefill
2026-05-02 20:51:23 +00:00 · 2025-09-13 21:54:08 -06:00
parent 3a51dbba85
commit e5d73da2da
2 changed files with 22 additions and 5 deletions
--- a/example/ck_tile/03_gemm/gemm_weight_preshuffle_invoker.hpp
+++ b/example/ck_tile/03_gemm/gemm_weight_preshuffle_invoker.hpp
@@ -106,7 +106,10 @@ struct WeightPreshuffleInvoker
                                                 GemmConfig::K_Warp_Tile,
                                                 UniversalGemmProblem::TransposeC,
                                                 memory_operation,
-                                                 GemmConfig::NumWaveGroups>>;
+                                                 GemmConfig::NumWaveGroups,
+                                                 false,
+                                                 1,
+                                                 true>>;
            using Kernel = ck_tile::GemmKernel<TilePartitioner, GemmPipeline, GemmEpilogue>;
            auto kargs   = Kernel::MakeKernelArgs(args);