执行摘要

• 一句话：修复 fused EH norm CI 注册参数
• 推荐动作：简单必要的修复，无需深入精读。但可作为示例说明 CI 配置变更模式。

功能与动机

修复 PR #29667 引入的 CI 注册问题——register_cuda_ci 的 suite 参数已被弃用，需要拆分为 stage 和 runner_config，否则 CI 会失败。

实现拆解

1. 修改 bench_fused_eh_norm.py：将 register_cuda_ci(est_time=6, suite="base-b-kernel-benchmark-1-gpu-large") 改为 register_cuda_ci(est_time=6, stage="base-b-kernel-benchmark", runner_config="1-gpu-large")。
2. 修改 test_fused_eh_norm.py：将 register_cuda_ci(est_time=45, suite="base-b-kernel-unit-1-gpu-large") 改为 register_cuda_ci(est_time=45, stage="base-b-kernel-unit", runner_config="1-gpu-large")。
3. 两个文件的其余逻辑未变，仅 CI 注册接口适配。

关键文件：

• test/registered/jit/benchmark/bench_fused_eh_norm.py（模块 基准测试；类别 test；类型 test-coverage）: 基准测试 CI 注册参数从 suite 改为 stage + runner_config。
• test/registered/jit/test_fused_eh_norm.py（模块 单元测试；类别 test；类型 test-coverage）: 单元测试 CI 注册参数从 suite 改为 stage + runner_config。

关键符号：未识别

关键源码片段

test/registered/jit/benchmark/bench_fused_eh_norm.py

基准测试 CI 注册参数从 suite 改为 stage + runner_config。

# 修改关键位置：CI 注册参数适配
from sglang.test.ci.ci_register import register_cuda_ci
​
# 原调用（已弃用）：
# register_cuda_ci(est_time=6, suite="base-b-kernel-benchmark-1-gpu-large")
# 新调用：
register_cuda_ci(
    est_time=6,
    stage="base-b-kernel-benchmark", # 替换 suite
    runner_config="1-gpu-large" # 独立运行器配置
)

test/registered/jit/test_fused_eh_norm.py

单元测试 CI 注册参数从 suite 改为 stage + runner_config。

# 修改关键位置：CI 注册参数适配
from sglang.test.ci.ci_register import register_cuda_ci
​
# 原调用（已弃用）：
# register_cuda_ci(est_time=45, suite="base-b-kernel-unit-1-gpu-large")
# 新调用：
register_cuda_ci(
    est_time=45,
    stage="base-b-kernel-unit", # 替换 suite
    runner_config="1-gpu-large" # 独立运行器配置
)
​
# nightly 注册保持不变
register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)

评论区精华

无人工 review 讨论。仅 bot 评论确认变更正确，无额外反馈。

• 暂无高价值评论线程

风险与影响

• 风险：变更仅涉及 CI 注册参数，不影响 kernel 逻辑或测试功能。若 stage 和 runner_config 在新的 CI 系统中不符合预期，可能导致基准测试或单元测试被跳过或运行失败，但概率低。
• 影响：影响仅限 CI 测试注册，确保 fused EH norm 相关的基准测试和单元测试能正确加入 CI 流水线。对用户无直接影响。
• 风险标记：CI 配置变更

关联脉络

• PR #29667 Add fused EH norm for DeepSeek NextN: 本 PR 修复了 #29667 中引入的 CI 注册问题。