From 6fb2b54ff4e827ee1fa42e0db73625bf4f22ddc3 Mon Sep 17 00:00:00 2001 From: OscarXu Date: Sat, 17 May 2025 01:03:28 -0500 Subject: [PATCH] mx_fp4 default parameter change --- example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4.cpp b/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4.cpp index f683c5075f..00e2ce3870 100644 --- a/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4.cpp +++ b/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4.cpp @@ -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)