ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 00:40:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

clean up example a bit

Browse Source
This commit is contained in:
Sami Remes
2026-02-06 18:07:36 +00:00
parent c588a1fd42
commit 241ee59880
3 changed files with 35 additions and 272 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
example/ck_tile/42_mx_gemm/mx_gemm.hpp
View File

@@ -73,9 +73,9 @@ struct MxGemmConfig
};
struct MXfp4_GemmConfig16 : MxGemmConfig
{
static constexpr ck_tile::index_t M_Tile = 32;
static constexpr ck_tile::index_t M_Tile = 64;
static constexpr ck_tile::index_t N_Tile = 64;
static constexpr ck_tile::index_t K_Tile = 512;
static constexpr ck_tile::index_t K_Tile = 256;
};
// GEMM config with 16x16 warp tile
@@ -83,5 +83,5 @@ struct MXfp8_GemmConfig16 : MxGemmConfig
{
static constexpr ck_tile::index_t M_Tile = 32;
static constexpr ck_tile::index_t N_Tile = 64;
static constexpr ck_tile::index_t K_Tile = 512;
static constexpr ck_tile::index_t K_Tile = 256;
};