From 6ab1de129ff5ce725a135af887ea9b5567d09a30 Mon Sep 17 00:00:00 2001 From: huizzhan Date: Tue, 26 Aug 2025 09:55:50 +0000 Subject: [PATCH] best perf pw gemm config 0.11ms --- example/ck_tile/03_gemm/gemm_utils.hpp | 4 ++-- example/ck_tile/03_gemm/gemm_weight_preshuffle.cpp | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) mode change 100644 => 100755 example/ck_tile/03_gemm/gemm_utils.hpp mode change 100644 => 100755 example/ck_tile/03_gemm/gemm_weight_preshuffle.cpp diff --git a/example/ck_tile/03_gemm/gemm_utils.hpp b/example/ck_tile/03_gemm/gemm_utils.hpp old mode 100644 new mode 100755 index ed2006d4b9..876f643988 --- a/example/ck_tile/03_gemm/gemm_utils.hpp +++ b/example/ck_tile/03_gemm/gemm_utils.hpp @@ -280,8 +280,8 @@ struct GemmConfigPreshuffleDecode : public GemmConfigBase template struct GemmConfigPreshufflePrefill : public GemmConfigBase { - static constexpr ck_tile::index_t M_Tile = 128; - static constexpr ck_tile::index_t N_Tile = 128; + static constexpr ck_tile::index_t M_Tile = 64; + static constexpr ck_tile::index_t N_Tile = 256; static constexpr ck_tile::index_t K_Tile = 128 / sizeof(PrecType); static constexpr ck_tile::index_t M_Warp = 1; diff --git a/example/ck_tile/03_gemm/gemm_weight_preshuffle.cpp b/example/ck_tile/03_gemm/gemm_weight_preshuffle.cpp old mode 100644 new mode 100755 index 0018db2c99..0843dba0dd --- a/example/ck_tile/03_gemm/gemm_weight_preshuffle.cpp +++ b/example/ck_tile/03_gemm/gemm_weight_preshuffle.cpp @@ -277,7 +277,7 @@ int main(int argc, char* argv[]) try { - return !run_gemm_example(arg_parser); + return !run_gemm_example(arg_parser); } catch(const std::runtime_error& e) {