From d1dbc69eda56fce085967c2ea1d6ba418a77776c Mon Sep 17 00:00:00 2001 From: mhYang Date: Mon, 31 Mar 2025 23:18:22 +0000 Subject: [PATCH] Use mfma 16x16x32 --- example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp b/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp index be78849148..1458fd997c 100644 --- a/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp +++ b/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp @@ -71,9 +71,9 @@ struct GridGemm static constexpr index_t M_Warp = 2; static constexpr index_t N_Warp = 2; static constexpr index_t K_Warp = 1; - static constexpr index_t M_Warp_Tile = 32; - static constexpr index_t N_Warp_Tile = 32; - static constexpr index_t K_Warp_Tile = 16; + static constexpr index_t M_Warp_Tile = 16; + static constexpr index_t N_Warp_Tile = 16; + static constexpr index_t K_Warp_Tile = 32; static constexpr bool DoubleSmemBuffer = false; static constexpr bool kPadM = false; static constexpr bool kPadN = false;