[Bugfix] Handle DeepseekV4ForCausalLM in benchmark_moe get_model_params (#52044)
Co-authored-by: Do_it_now_! <23432123@users.noreply.github.com>
Do_it_now_! committed
Aug 18, 2026 at 12:46 UTC
101c4477dd4538f6fc32161d680ff2fba942b6d0
1 file changed
+1
benchmarks/kernels/benchmark_moe.py
+1
@@ -766,6 +766,7 @@ def get_model_params(config):
766
"DeepseekV2ForCausalLM",
767
"DeepseekV3ForCausalLM",
768
"DeepseekV32ForCausalLM",
769
+ "DeepseekV4ForCausalLM",
770
"GlmMoeDsaForCausalLM",
771
"Glm4MoeForCausalLM",
772
"Glm4MoeLiteForCausalLM",