Skip to content

Commit 67f3fb0

Browse files
authored
[Bench] Add DeepSeekV32 to MoE benchmark (#25962)
Signed-off-by: Jee Jee Li <[email protected]>
1 parent 43b752c commit 67f3fb0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

benchmarks/kernels/benchmark_moe.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -584,8 +584,9 @@ def main(args: argparse.Namespace):
584584
topk = config.num_experts_per_tok
585585
intermediate_size = config.intermediate_size
586586
elif config.architectures[0] in (
587-
"DeepseekV3ForCausalLM",
588587
"DeepseekV2ForCausalLM",
588+
"DeepseekV3ForCausalLM",
589+
"DeepseekV32ForCausalLM",
589590
"Glm4MoeForCausalLM",
590591
):
591592
E = config.n_routed_experts

0 commit comments

Comments
 (0)