执行摘要

• 一句话：修复 FusedMoE 专家映射调用点
• 推荐动作：建议精读，这是一个教科书级的重构遗漏修复，适合理解跨文件重构的收尾工作。

功能与动机

在 #41184 重构中，FusedMoE.make_expert_params_mapping 被重构为独立的 fused_moe_make_expert_params_mapping 函数，但部分调用点未被更新。PR body 明确指出：‘Fixes stale fused MoE expert mapping call sites after #41184, restoring pre-commit on main and avoiding runtime failures in affected weight-loading paths.’

实现拆解

1. 导入调整：在 6 个文件中，将 from vllm.model_executor.layers.fused_moe import FusedMoE 改为同时导入 fused_moe_make_expert_params_mapping（或仅导入新函数）。
2. 调用替换：将所有 FusedMoE.make_expert_params_mapping(...) 替换为 fused_moe_make_expert_params_mapping(...)，参数签名保持不变。
3. 注释更新：在 vllm/lora/model_manager.py 中更新了一处注释，将 FusedMoE.make_expert_params_mapping 改为 fused_moe_make_expert_params_mapping，确保文档与代码一致。

关键文件：

• vllm/model_executor/models/hy_v3.py（模块 模型加载；类别 source；类型 data-contract；符号 get_expert_mapping）: 主模型文件，替换了 FusedMoE.make_expert_params_mapping 调用。
• vllm/models/deepseek_v4/xpu/model.py（模块 模型加载；类别 source；类型 data-contract；符号 get_expert_mapping）: DeepSeek V4 XPU 模型，同样需要更新调用。
• vllm/model_executor/models/hy_v3_mtp.py（模块 模型加载；类别 source；类型 data-contract；符号 load_weights）: HY V3 MTP 模型，更新权重加载中的参数映射。
• vllm/models/deepseek_v4/xpu/mtp.py（模块 模型加载；类别 source；类型 data-contract；符号 load_weights）: DeepSeek V4 XPU MTP 模型，更新权重加载中的参数映射。
• vllm/model_executor/models/cohere2_moe.py（模块 模型加载；类别 source；类型 data-contract；符号 load_weights）: Cohere2 MoE 模型，更新权重加载中的参数映射。
• vllm/lora/model_manager.py（模块 LoRA；类别 source；类型 data-contract；符号 _restrict_to_local_experts）: LoRA 模型管理器，注释调整，但无逻辑变更。

关键符号：get_expert_mapping, load_weights, _restrict_to_local_experts

关键源码片段

vllm/model_executor/models/hy_v3.py

主模型文件，替换了 FusedMoE.make_expert_params_mapping 调用。

# vllm/model_executor/models/hy_v3.py (partial)
​
def get_expert_mapping(self) -> list[tuple[str, str, int, str]]:
    # 返回 expert params mapping 列表，用于权重加载
    # 重构后使用独立的函数取代 FusedMoE 类方法
    return fused_moe_make_expert_params_mapping(
        self,
        ckpt_gate_proj_name="gate_proj",
        ckpt_down_proj_name="down_proj",
        ckpt_up_proj_name="up_proj",
        num_experts=self.config.num_experts,
    )

评论区精华

无实质性讨论，仅自动审核评论。

• 暂无高价值评论线程

风险与影响

• 风险：风险极低。变更仅为函数调用重定向，参数签名完全一致，且经过 CI 验证。但若未来 fused_moe_make_expert_params_mapping 的语义发生变化，这些调用点需要同步更新。
• 影响：直接修复了 main 分支上因重构导致的 pre-commit 阻塞和潜在的权重加载崩溃，影响所有使用 MoE 的模型（如 deepseek、cohere、hy v3 等）。
• 风险标记：跨文件变更, 低风险

关联脉络

• PR #41184 [MoE Refactor] FusedMoE/MoERunner inversion refactor: 本 PR 修复了 #41184 重构中遗留的调用点未更新问题