ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

mx_fp4 default parameter change

Browse Source
This commit is contained in:
OscarXu
2025-05-17 01:03:28 -05:00
parent ec8d00d58d
commit 6fb2b54ff4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4.cpp
View File

@@ -189,14 +189,14 @@ int main(int argc, char* argv[])
// per expert:
// GEMM shape
ck::index_t N = 14336;
ck::index_t N = 6144;
ck::index_t K = 4096;
ck::index_t experts = 1;
ck::index_t experts = 8;
ck::index_t sorted_tile_num = 16;
ck::index_t valid_tile_num = 13;
ck::index_t sorted_size = sorted_tile_num * MPerBlock;
ck::index_t valid_size = valid_tile_num * MPerBlock;
ck::index_t tokens = 644;
ck::index_t tokens = 832;
ck::index_t topk = 2;
if(argc == 1)