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

enable fp8 mx gemm too

Browse Source
This commit is contained in:
Sami Remes
2026-01-30 12:43:49 -05:00
parent 771c46aa8b
commit b8cdea5979
2 changed files with 13 additions and 13 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

@@ -81,7 +81,7 @@ struct MXfp4_GemmConfig16 : MxGemmConfig
// GEMM config with 16x16 warp tile
struct MXfp8_GemmConfig16 : MxGemmConfig
{
static constexpr ck_tile::index_t M_Tile = 128;
static constexpr ck_tile::index_t N_Tile = 128;
static constexpr ck_tile::index_t K_Tile = 256;
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;
};