[BugFix] Align fused MoE-LoRA kernel config with actual weight shapes (#34396)

Signed-off-by: Runkai Tao <rt572@physics.rutgers.edu>
This commit is contained in:
Runkai Tao
2026-02-26 13:03:10 -05:00
committed by GitHub
parent 05970c772c
commit a1f53addb1

View File

@@ -83,7 +83,11 @@ class FusedMoEWithLoRA(BaseLayerWithLoRA):
):
if envs.VLLM_TUNED_CONFIG_FOLDER:
hidden_size = layer.hidden_size
intermediate_size = layer.intermediate_size_per_partition
intermediate_size = (
self.w2_lora_a_stacked[0].shape[-1]
if op_prefix == "w2"
else self.w13_lora_b_stacked[0].shape[-2]
)
shrink_config = get_lora_op_configs(
op_type=f"fused_moe_lora_{op_prefix}_shrink",
max_loras=num_loras,