ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-03 21:58:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use mfma 16x16x32

Browse Source
This commit is contained in:
mhYang
2025-03-31 23:18:22 +00:00
parent ee28e965f2
commit d1dbc69eda
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp
View File

@@ -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;