From 397aeca3762478161c2e11021e89d535f1fb6ae6 Mon Sep 17 00:00:00 2001 From: Xiaoyu Zhang <1182563586@qq.com> Date: Thu, 3 Sep 2026 17:21:42 +0800 Subject: [PATCH] fix(benchmark): support Glm4MoeLite in fused MoE tuner (#37623) --- benchmark/kernels/fused_moe_triton/common_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/benchmark/kernels/fused_moe_triton/common_utils.py b/benchmark/kernels/fused_moe_triton/common_utils.py index 4c8b4625a..b5554a090 100644 --- a/benchmark/kernels/fused_moe_triton/common_utils.py +++ b/benchmark/kernels/fused_moe_triton/common_utils.py @@ -96,6 +96,7 @@ def get_model_config( "DeepseekV32ForCausalLM", "DeepseekV4ForCausalLM", "Glm4MoeForCausalLM", + "Glm4MoeLiteForCausalLM", "GlmMoeDsaForCausalLM", "KimiVLForConditionalGeneration", "MistralLarge3ForCausalLM", @@ -108,6 +109,7 @@ def get_model_config( "DeepseekV3ForCausalLM", "DeepseekV32ForCausalLM", "Glm4MoeForCausalLM", + "Glm4MoeLiteForCausalLM", "GlmMoeDsaForCausalLM", "MistralLarge3ForCausalLM", ]