From 2de3e71d17e0197e849d4ecc0c2ed6a7004708b0 Mon Sep 17 00:00:00 2001 From: wangzaijun Date: Fri, 10 Jul 2026 00:27:31 +0000 Subject: [PATCH] fix moe tuning configs --- .../basemodel/triton_kernel/fused_moe/grouped_fused_moe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]