diff --git a/example/ck_tile/03_gemm/gemm_utils.hpp b/example/ck_tile/03_gemm/gemm_utils.hpp index 588b66ca43..07b925d0eb 100644 --- a/example/ck_tile/03_gemm/gemm_utils.hpp +++ b/example/ck_tile/03_gemm/gemm_utils.hpp @@ -72,6 +72,7 @@ struct GemmConfigBase static constexpr ck_tile::index_t Pipeline = CK_TILE_PIPELINE_COMPUTE_V3; static constexpr ck_tile::index_t NumWaveGroups = 1; static constexpr bool Preshuffle = false; + static constexpr bool TiledMMAPermuteN = false; }; template diff --git a/example/ck_tile/03_gemm/gemm_weight_preshuffle_invoker.hpp b/example/ck_tile/03_gemm/gemm_weight_preshuffle_invoker.hpp index b47dd8d8a7..d737a0f864 100644 --- a/example/ck_tile/03_gemm/gemm_weight_preshuffle_invoker.hpp +++ b/example/ck_tile/03_gemm/gemm_weight_preshuffle_invoker.hpp @@ -109,7 +109,7 @@ struct WeightPreshuffleInvoker GemmConfig::NumWaveGroups, false, 1, - true>>; + GemmConfig::TiledMMAPermuteN>>; using Kernel = ck_tile::GemmKernel; auto kargs = Kernel::MakeKernelArgs(args);