执行摘要

• 一句话：删除 topk_softplus_sqrt 的 XPU Torch 分支，统一走 vllm 算子
• 推荐动作：这是一个值得快速阅读 diff 的小型清理 PR，核心价值在于展示“平台专属 fallback 下沉到 vendor kernel 库后删除分支”的演进模式，对后续 XPU 算子迁移类 PR 有参考意义。建议合入前确认 vllm_xpu_kernels 的可用版本，并在后续补充一个最小 XPU 回归测试以覆盖该路由路径。

功能与动机

PR body 说明：“The topk_softplus_sqrt operator has now been merged into vllm_xpu_kernels, so we can remove the XPU Torch branch.” 即 XPU 平台的专属算子已下沉到 vllm_xpu_kernels，原先为了在 XPU 上运行而保留的纯 PyTorch fallback 不再必要，从而收敛分支、统一走自定义算子路径。

实现拆解

1. 定位文件 vllm/model_executor/layers/fused_moe/router/fused_topk_bias_router.py，这是本次唯一变更文件。
2. 删除 _topk_softplus_sqrt_torch 函数：该函数是 XPU/CPU 的纯 PyTorch fallback，包含 sqrt(softplus(gating_output)) 评分、e_score_correction_bias 仅用于选专家（权重使用无偏得分）、hash MoE 查表、renormalize、routed_scaling_factor 应用等完整逻辑。
3. 简化 vllm_topk_softplus_sqrt：移除 current_platform.is_xpu() 分支与 from vllm.platforms import current_platform 导入，函数体不再分叉，直接调用 ops.topk_hash_softplus_sqrt。
4. 移除 import torch.nn.functional as F：F 仅被删除的 fallback 使用。
5. 测试与构建：本 PR 未新增或修改测试文件，XPU 行为验证依赖 vllm_xpu_kernels 侧实现与 Buildkite CI #82655（经一次 retry 后通过）。

关键文件：

• vllm/model_executor/layers/fused_moe/router/fused_topk_bias_router.py（模块 MoE 路由；类别 source；类型 cleanup；符号 _topk_softplus_sqrt_torch, vllm_topk_softplus_sqrt）: 唯一变更文件：删除 _topk_softplus_sqrt_torch fallback 实现和 vllm_topk_softplus_sqrt 中的 XPU 分支判断，统一调用 ops.topk_hash_softplus_sqrt。

关键符号：vllm_topk_softplus_sqrt, _topk_softplus_sqrt_torch

关键源码片段

vllm/model_executor/layers/fused_moe/router/fused_topk_bias_router.py

唯一变更文件：删除 _topk_softplus_sqrt_torch fallback 实现和 vllm_topk_softplus_sqrt 中的 XPU 分支判断，统一调用 ops.topk_hash_softplus_sqrt。

def vllm_topk_softplus_sqrt(
    topk_weights: torch.Tensor,
    topk_indices: torch.Tensor,
    token_expert_indices: torch.Tensor,
    gating_output: torch.Tensor,
    renormalize: bool = False,
    e_score_correction_bias: torch.Tensor | None = None,
    input_tokens: torch.Tensor | None = None,
    hash_indices_table: torch.Tensor | None = None,
    routed_scaling_factor: float = 1.0,
) -> tuple[torch.Tensor, ...]:
    # MoE 路由的 softplus + sqrt 评分 + topk 选择统一入口。
    # 原 XPU 分支已移除（PR #51242）：topk_softplus_sqrt 算子合入
    # vllm_xpu_kernels 后，所有平台统一调用 ops.topk_hash_softplus_sqrt，
    # 不再需要维护纯 PyTorch 的 fallback 实现。
    ops.topk_hash_softplus_sqrt(
        topk_weights,
        topk_indices,
        token_expert_indices,
        gating_output,
        renormalize,
        routed_scaling_factor,
        e_score_correction_bias,
        input_tokens,
        hash_indices_table,
        is_padding=_get_padding_mask(topk_indices.shape[0]),
    )
​
    return topk_weights, topk_indices

评论区精华

公开 review 评论为空，核心交流集中在 CI 与合入流程：

• claude[bot] 指出该 PR 来自 fork，自动 review 被禁用，维护者可手动触发。
• jikunshang 触发 /ci run，2 个失败任务经 /ci retry 重试后通过。

• 最终 jikunshang 无文字批注直接 APPROVED。
  没有针对删除行为本身的公开技术讨论；相对值得关注的未决点：删除 fallback 后 XPU 行为完全取决于 vllm_xpu_kernels 的算子实现，仓库内无直接测试覆盖说明。

• CI 验证与合入决策 (other): 无代码层面的公开技术讨论；合并主要依赖 CI 通过和维护者的审核。

风险与影响

• 风险：
  1. 外部内核依赖：XPU 运行路径从纯 PyTorch 切换为 vllm_xpu_kernels 中的 topk_hash_softplus_sqrt，若用户环境的 vllm_xpu_kernels 版本未包含该算子，将在 MoE 路由阶段找不到符号而报错。
  2. 数值行为一致性：fallback 中 bias 只参与 expert 选择、权重使用无偏得分，renormalize 使用 clamp(min=1e-20)；若 vllm_xpu_kernels 实现细节不同，XPU 上 MoE 路由结果可能发生细微变化（例如 DSv4-Flash 场景偏置接近均匀时）。
  3. 测试覆盖不足：PR 未附带针对该路径的单元测试，若未来其他平台也需要 fallback，需要重新引入并验证。
     - 影响：影响范围集中在 XPU 平台的 MoE 路由路径（fused_topk_bias_router.py）。对用户而言，XPU 推理路径实现切换但预期行为一致，且可能获得内核级性能提升；对系统而言，消除了双实现漂移风险，未来维护只需跟进 vllm_xpu_kernels；对团队而言，Intel XPU 维护者需要确保 vllm_xpu_kernels 的发布与 vLLM 版本兼容绑定。
     - 风险标记：外部内核库依赖, 缺少专项测试, 数值行为一致性风险

关联脉络

• PR #50840 [XPU] Route AWQ linear through choose_mp_linear_kernel: 同类 XPU 收敛清理：删除专属实现、统一走公共内核选择路径，与本 PR 的“移除 XPU 分支、统一算子入口”模式一致。
• PR #50946 [XPU] Register fake meta kernel for fp4_gemm: 同为 XPU 算子与 vLLM 内核集成相关的适配，反映 XPU 算子持续下沉到 XPU 专用内核库/注册体系的演进趋势。