执行摘要

• 一句话：升级 Helion 至 1.4.0 并同步 CI，修复内核数值问题
• 推荐动作：该 PR 是简单合规的依赖升级，不必精读，但值得注意其揭示的维护模式：一处依赖版本需要三处同步，且同步完全依赖代码注释提醒，缺少自动化校验。可考虑在未来引入脚本或 CI 检查来防止版本漂移，这是比本次升级本身更有价值的改进点。

功能与动机

PR body 明确说明升级目的是获取 pytorch/helion#3081 的修复，该修复解决了 fused_qk_norm_rope Helion 内核中的数值问题；同时 setup.py 注释要求升级 Helion 时同步更新两个 CI 文件，本次一并完成。

实现拆解

1. 更新主依赖定义（setup.py）：在 extras_require 中把 "helion": ["helion==1.1.0"] 改为 "helion": ["helion==1.4.0"]。该键为可选依赖，不影响默认安装路径。
2. 同步 NVIDIA CI 配置（.buildkite/test_areas/kernels.yaml）：kernels-helion-test（H100，并行度 2）中的 pip install 行从 1.1.0 改为 1.4.0，保证 CI 与安装依赖版本一致。
3. 同步 AMD CI 配置（.buildkite/test-amd.yaml）：MI250 上的 Kernels Helion Test 同样将 pip install 更新为 1.4.0，避免 ROCm 侧测试跑在旧版本上。
4. 验证与收尾：未新增单元测试，依赖 tests/kernels/helion/ 下的既有测试集验证内核数值正确性。作者 rebase 后 auto-merge CI 曾失败，但确认是 test_chat_streaming_audio 的已知 flake（issue #50385），与本 PR 无关；最终由 zou3519 和 xiaohongchen1991 批准合并。

关键文件：

• setup.py（模块 构建配置；类别 source；类型 core-logic）: 额外依赖 helion 的版本定义所在，是本次升级的源头；其注释明确要求同步更新两个 CI 文件，构成三处同步的基础。
• .buildkite/test-amd.yaml（模块 CI 配置；类别 config；类型 configuration）: MI250 上 Kernels Helion Test 的安装命令同步升级，避免 ROCm 侧测试使用旧版本 Helion。
• .buildkite/test_areas/kernels.yaml（模块 CI 配置；类别 config；类型 configuration）: H100 上 kernels-helion-test 的依赖安装命令同步升级，是 NVIDIA 侧 Helion 内核测试的版本锚点。

关键符号：未识别

关键源码片段

setup.py

额外依赖 helion 的版本定义所在，是本次升级的源头；其注释明确要求同步更新两个 CI 文件，构成三处同步的基础。

# setup.py —— extras_require 中的 helion 可选依赖（1.1.0 → 1.4.0）
setup(
    ...
    extras_require={
        ...
        # Helion 内核开发可选依赖
        # NOTE: 升级 Helion 版本时，必须同步更新：
        # - .buildkite/test_areas/kernels.yaml
        # - .buildkite/test-amd.yaml
        "helion": ["helion==1.4.0"], # 1.4.0 修复 fused_qk_norm_rope 数值问题，见 pytorch/helion#3081
        ...
    },
)

评论区精华

作者 yushangdi 在评论中说明做了 rebase，部分 auto-merge CI 失败；随后补充 "The failures are unrelated to the Helion pin"，并指出 test_chat_streaming_audio 在多个无关 PR 上以相同方式失败（issue #50385）。claude[bot] 提示该 PR 来自 fork，自动 code review 被禁用，维护者可手动 @claude review；实际由 zou3519 与 xiaohongchen1991 两人直接批准，无额外 review 评论。

• rebase 后 auto-merge CI 失败 (testing): 作者判断失败与 Helion 版本 pin 无关，属于已知 flake。
• test_chat_streaming_audio 已知 flake（issue #50385） (testing): CI 失败与本 PR 无关，PR 最终合并。
• fork 仓库自动 code review 禁用 (other): 由人类 reviewer zou3519、xiaohongchen1991 直接批准。

风险与影响

• 风险：
  • 数值行为变更：fused_qk_norm_rope 修复会改变内核输出数值。对于已受旧版本数值问题影响的使用者，升级后输出可能发生变化（这正是修复目标），若存在依赖旧输出的对齐测试则需关注。
  • 多文件版本同步：Helion 版本号同时存在于 setup.py 与两个 CI yaml 中，本次手工同步完成，但缺少自动化校验，未来升级存在漏改风险。
  • 影响面：仅作用于显式安装 vllm[helion] 或运行 kernels/helion/ CI 测试的 Helion 路径，默认安装与主流推理路径不受影响；若 1.4.0 引入新依赖或 API 变化，CI 测试会第一时间暴露。
  • 影响：影响范围较小：仅涉及 Helion 可选依赖与两组 CI 任务（H100 与 MI250）。受益方是使用 Helion fused_qk_norm_rope 内核的用户，数值准确性得到提升；对默认安装与主流 NVIDIA/AMD 推理路径无影响。团队维护上，后续 Helion 升级可沿用此 PR 建立的“setup.py + 两处 CI yaml”三文件同步模式。
  • 风险标记：依赖版本升级, 多文件版本同步, 数值行为变更

关联脉络

• 暂无明显关联 PR