From 751c4540990152bc045d44859ca1498d34337750 Mon Sep 17 00:00:00 2001
From: Yuan Luo <yuan.luo@hotmail.com>
Date: Tue, 10 Mar 2026 17:12:58 +0800
Subject: [PATCH] Add DeepSeek3.2 and GlmMoeDsa into moe tune (#18876)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
---
 benchmark/kernels/fused_moe_triton/common_utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/benchmark/kernels/fused_moe_triton/common_utils.py b/benchmark/kernels/fused_moe_triton/common_utils.py
index adac313a1..b7d5b4cc6 100644
--- 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