Add DeepSeek3.2 and GlmMoeDsa into moe tune (#18876)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
2026-03-10 17:12:58 +08:00
parent ac07a6d439
commit 751c454099
1 changed files with 4 additions and 0 deletions
--- a/benchmark/kernels/fused_moe_triton/common_utils.py
+++ b/benchmark/kernels/fused_moe_triton/common_utils.py
@@ -86,7 +86,9 @@ def get_model_config(
    elif architecture in [
        "DeepseekV2ForCausalLM",
        "DeepseekV3ForCausalLM",
+        "DeepseekV32ForCausalLM",
        "Glm4MoeForCausalLM",
+        "GlmMoeDsaForCausalLM",
        "MistralLarge3ForCausalLM",
    ]:
        E = (config.n_routed_experts // ep_size) + (
@@ -95,7 +97,9 @@ def get_model_config(
            or architecture
            not in [
                "DeepseekV3ForCausalLM",
+                "DeepseekV32ForCausalLM",
                "Glm4MoeForCausalLM",
+                "GlmMoeDsaForCausalLM",
                "MistralLarge3ForCausalLM",
            ]
            else 1