From e65a7163675f4be90ee949b824a0e5caf06c0903 Mon Sep 17 00:00:00 2001 From: AviralGoelAMD Date: Fri, 18 Jul 2025 23:00:55 +0000 Subject: [PATCH] update kwarp calculation in gemm utils --- example/ck_tile/03_gemm/gemm_utils.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/03_gemm/gemm_utils.hpp b/example/ck_tile/03_gemm/gemm_utils.hpp index eed7423d9a..98db148e82 100644 --- a/example/ck_tile/03_gemm/gemm_utils.hpp +++ b/example/ck_tile/03_gemm/gemm_utils.hpp @@ -289,11 +289,11 @@ struct GemmConfigPreshuffle_3 : public GemmConfigBase static constexpr ck_tile::index_t M_Warp_Tile = 16; static constexpr ck_tile::index_t N_Warp_Tile = 16; - static constexpr ck_tile::index_t K_Warp_Tile = 32; + static constexpr ck_tile::index_t K_Warp_Tile = get_k_warp_tile_flatmm(); static constexpr int kBlockPerCu = 2; static constexpr auto Scheduler = ck_tile::GemmPipelineScheduler::Default; - static constexpr ck_tile::index_t Pipeline = CK_TILE_PIPELINE_PRESHUFFLE_V2; + static constexpr ck_tile::index_t Pipeline = CK_TILE_PIPELINE_PRESHUFFLE_V3; static constexpr bool Preshuffle = true; static constexpr bool DoubleSmemBuffer = true; };