From d11ec124a0409a78a23539212356510c79c2111e Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Thu, 11 Sep 2025 21:29:43 +0800 Subject: [PATCH] [Bench] Add qwen-next in benchmark_moe.py (#24661) Signed-off-by: Jee Jee Li --- benchmarks/kernels/benchmark_moe.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/benchmarks/kernels/benchmark_moe.py b/benchmarks/kernels/benchmark_moe.py index 6259aa0dd629..94f3f1ae11f2 100644 --- a/benchmarks/kernels/benchmark_moe.py +++ b/benchmarks/kernels/benchmark_moe.py @@ -594,7 +594,11 @@ def main(args: argparse.Namespace): E = config.n_routed_experts topk = config.num_experts_per_tok intermediate_size = config.moe_intermediate_size - elif config.architectures[0] in ("Qwen2MoeForCausalLM", "Qwen3MoeForCausalLM"): + elif config.architectures[0] in ( + "Qwen2MoeForCausalLM", + "Qwen3MoeForCausalLM", + "Qwen3NextForCausalLM", + ): E = config.num_experts topk = config.num_experts_per_tok intermediate_size = config.moe_intermediate_size