From ef318228e7de8af061cc4b0c6ac56c8ba98936a0 Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Wed, 1 Oct 2025 05:13:48 +0800 Subject: [PATCH] [Bench] Add DeepSeekV32 to MoE benchmark (#25962) Signed-off-by: Jee Jee Li Signed-off-by: yewentao256 --- benchmarks/kernels/benchmark_moe.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/benchmarks/kernels/benchmark_moe.py b/benchmarks/kernels/benchmark_moe.py index d2beb28f70233..02c2db674d4b0 100644 --- 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