[Bench] Add DeepSeekV32 to MoE benchmark (#25962)

Signed-off-by: Jee Jee Li <pandaleefree@gmail.com> Signed-off-by: simon-mo <simon.mo@hey.com>
2025-10-01 05:13:48 +08:00
parent 83f3c9beae
commit c3dfb0f6dd
1 changed files with 2 additions and 1 deletions
--- a/benchmarks/kernels/benchmark_moe.py
+++ b/benchmarks/kernels/benchmark_moe.py
@ -584,8 +584,9 @@ def main(args: argparse.Namespace):
        topk = config.num_experts_per_tok
        intermediate_size = config.intermediate_size
    elif config.architectures[0] in (
-        "DeepseekV3ForCausalLM",
        "DeepseekV2ForCausalLM",
+        "DeepseekV3ForCausalLM",
+        "DeepseekV32ForCausalLM",
        "Glm4MoeForCausalLM",
    ):
        E = config.n_routed_experts