diff --git a/benchmark/kernels/fused_moe_triton/common_utils.py b/benchmark/kernels/fused_moe_triton/common_utils.py index adac313a1..b7d5b4cc6 100644 --- a/benchmark/kernels/fused_moe_triton/common_utils.py +++ b/benchmark/kernels/fused_moe_triton/common_utils.py @@ -86,7 +86,9 @@ def get_model_config( elif architecture in [ "DeepseekV2ForCausalLM", "DeepseekV3ForCausalLM", + "DeepseekV32ForCausalLM", "Glm4MoeForCausalLM", + "GlmMoeDsaForCausalLM", "MistralLarge3ForCausalLM", ]: E = (config.n_routed_experts // ep_size) + ( @@ -95,7 +97,9 @@ def get_model_config( or architecture not in [ "DeepseekV3ForCausalLM", + "DeepseekV32ForCausalLM", "Glm4MoeForCausalLM", + "GlmMoeDsaForCausalLM", "MistralLarge3ForCausalLM", ] else 1