From 101c4477dd4538f6fc32161d680ff2fba942b6d0 Mon Sep 17 00:00:00 2001
From: Do_it_now_! <lizhengcheng233@gmail.com>
Date: Tue, 18 Aug 2026 12:46:08 +0800
Subject: [PATCH] [Bugfix] Handle DeepseekV4ForCausalLM in benchmark_moe
 get_model_params (#52044)

Co-authored-by: Do_it_now_! <23432123@users.noreply.github.com>
---
 benchmarks/kernels/benchmark_moe.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/benchmarks/kernels/benchmark_moe.py b/benchmarks/kernels/benchmark_moe.py
index 9ea22dd130..5d86a7599f 100644
--- a/benchmarks/kernels/benchmark_moe.py
+++ b/benchmarks/kernels/benchmark_moe.py
@@ -766,6 +766,7 @@ def get_model_params(config):
         "DeepseekV2ForCausalLM",
         "DeepseekV3ForCausalLM",
         "DeepseekV32ForCausalLM",
+        "DeepseekV4ForCausalLM",
         "GlmMoeDsaForCausalLM",
         "Glm4MoeForCausalLM",
         "Glm4MoeLiteForCausalLM",