@@ -518,14 +518,14 @@ void dispatch_moe_gemm_to_cutlass(const T* A,
518
518
switch (gemm_config.tile_config ) {
519
519
dispatch_gemm_config_macro (16 , 128 , 64 , 16 , 32 , 64 );
520
520
dispatch_gemm_config_macro (16 , 256 , 64 , 16 , 64 , 64 );
521
- dispatch_gemm_config_macro (64 , 64 , 64 , 32 , 32 , 64 );
522
521
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 );
524
523
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 );
525
526
dispatch_gemm_config_macro (128 , 128 , 64 , 64 , 64 , 64 );
526
527
dispatch_gemm_config_macro (128 , 128 , 64 , 128 , 32 , 64 );
527
528
dispatch_gemm_config_macro (128 , 256 , 64 , 64 , 64 , 64 );
528
- dispatch_gemm_config_macro (64 , 128 , 64 , 64 , 32 , 64 );
529
529
dispatch_gemm_config_macro (256 , 128 , 64 , 64 , 64 , 64 );
530
530
case CutlassTileConfig::Undefined:
531
531
throw std::runtime_error (
0 commit comments