diff --git a/lightllm/common/basemodel/triton_kernel/fused_moe/grouped_fused_moe.py b/lightllm/common/basemodel/triton_kernel/fused_moe/grouped_fused_moe.py index cbed964f2..e10adf775 100644 --- a/lightllm/common/basemodel/triton_kernel/fused_moe/grouped_fused_moe.py +++ b/lightllm/common/basemodel/triton_kernel/fused_moe/grouped_fused_moe.py @@ -760,7 +760,7 @@ def _get_grouped_matmul_configs(): } for ns in [2, 3, 4, 5] for gm in [1, 16, 32, 64] - for nw in [4, 8] + for nw in [2, 4, 8] for bm in [16, 32, 64, 128] for bn in [16, 32, 64, 128] for bk in [32, 64, 128]