We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 43b752c commit 67f3fb0Copy full SHA for 67f3fb0
benchmarks/kernels/benchmark_moe.py
@@ -584,8 +584,9 @@ def main(args: argparse.Namespace):
584
topk = config.num_experts_per_tok
585
intermediate_size = config.intermediate_size
586
elif config.architectures[0] in (
587
- "DeepseekV3ForCausalLM",
588
"DeepseekV2ForCausalLM",
+ "DeepseekV3ForCausalLM",
589
+ "DeepseekV32ForCausalLM",
590
"Glm4MoeForCausalLM",
591
):
592
E = config.n_routed_experts
0 commit comments