Skip to content

Commit 7c73417

Browse files
committed
update moe config
1 parent 99e1c10 commit 7c73417

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

csrc/gpu/moe/fused_moe/cutlass_kernels/moe_gemm/fused_moe_gemm_kernels_template.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -518,14 +518,14 @@ void dispatch_moe_gemm_to_cutlass(const T* A,
518518
switch (gemm_config.tile_config) {
519519
dispatch_gemm_config_macro(16, 128, 64, 16, 32, 64);
520520
dispatch_gemm_config_macro(16, 256, 64, 16, 64, 64);
521-
dispatch_gemm_config_macro(64, 64, 64, 32, 32, 64);
522521
dispatch_gemm_config_macro(32, 128, 64, 32, 32, 64);
523-
dispatch_gemm_config_macro(128, 64, 64, 64, 32, 64);
522+
dispatch_gemm_config_macro(64, 64, 64, 32, 32, 64);
524523
dispatch_gemm_config_macro(64, 128, 64, 64, 64, 64);
524+
dispatch_gemm_config_macro(64, 128, 64, 64, 32, 64);
525+
dispatch_gemm_config_macro(128, 64, 64, 64, 32, 64);
525526
dispatch_gemm_config_macro(128, 128, 64, 64, 64, 64);
526527
dispatch_gemm_config_macro(128, 128, 64, 128, 32, 64);
527528
dispatch_gemm_config_macro(128, 256, 64, 64, 64, 64);
528-
dispatch_gemm_config_macro(64, 128, 64, 64, 32, 64);
529529
dispatch_gemm_config_macro(256, 128, 64, 64, 64, 64);
530530
case CutlassTileConfig::Undefined:
531531
throw std::runtime_error(

0 commit comments

Comments
 (0)