Prhub

#29197 [AMD] Register 5 JIT kernel unit tests for AMD nightly CI

原始 PR 作者 michaelzhang-ai 合并时间 2026-06-26 04:12 文件变更 5 提交数 1 评论 1 代码增减 +10 / -5

执行摘要

为 5 个 JIT 单元测试注册 AMD 夜间 CI

为了将已在AMD GPU上验证通过的JIT内核单元测试纳入AMD夜间CI,确保AMD平台的持续正确性。PR body明确指出这些测试在ROCm 7.0和7.2上全部通过,且是#28967的下一批。

建议精读PR body中的验证表格和CI运行链接,了解哪些测试被确认兼容及资源消耗。这对于AMD平台开发者或有AMD部署需求的用户有参考价值。

讨论亮点

PR仅有一个来自HaiShaw的审核,状态为APPROVED,无review评论。无实质讨论。

实现拆解

  1. 导入 register_amd_ci:在5个测试文件中,将导入语句从 from sglang.test.ci.ci_register import register_cuda_ci 改为 from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci

  2. 添加注册调用:在每个文件已有的CUDA注册之后,添加一行 register_amd_ci(est_time=..., suite="nightly-amd-kernel-1-gpu", nightly=True),其中 est_time 根据各测试的实际运行时间设定(15-30秒不等)。

  3. 无需工作流修改:夜间AMD CI套件 nightly-amd-kernel-1-gpu 已在之前的PR中配置于 nightly-test-amd.ymlnightly-test-amd-rocm720.yml,因此本次变更无需修改CI工作流文件。

文件 模块 状态 重要度
test/registered/jit/deepseek_v4/test_c128_v2.py JIT 测试 modified 3.92
test/registered/jit/deepseek_v4/test_c4_v2.py JIT 测试 modified 3.92
test/registered/jit/diffusion/test_group_norm_silu.py JIT 测试 modified 3.92
test/registered/jit/diffusion/test_qwen_image_modulation.py JIT 测试 modified 3.92
test/registered/jit/diffusion/test_varlen_pack_pad.py JIT 测试 modified 3.92

关键源码片段

test/registered/jit/deepseek_v4/test_c128_v2.py test-coverage

DeepSeek V4 c128 压缩内核测试,加入 AMD CI 注册,确保 AMD 平台对此关键内核的回归测试。

from __future__ import annotationsimport sys
from typing import Tuple, Unionimport pytest
import torch
import tritonfrom sglang.jit_kernel.dsv4 import compress_forward
from sglang.jit_kernel.tests.deepseek_v4.common import (
    LegacyContext,
    PagedContext,
    make_legacy_context,
    make_paged_context,
    make_state_pool,
    to_seq_extend,
)
# 同时导入 register_amd_ci 和 register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci# 保留原有 CUDA 注册
register_cuda_ci(est_time=30, suite="base-b-kernel-unit-1-gpu-large")
register_cuda_ci(est_time=30, suite="nightly-kernel-1-gpu", nightly=True)
# 新增 AMD 夜间 CI 注册
register_amd_ci(est_time=30, suite="nightly-amd-kernel-1-gpu", nightly=True)

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

风险极低。变更纯为测试注册,不涉及任何生产代码逻辑修改。每个文件仅修改2行(导入+注册),且已在两个ROCm版本上验证通过。存在极小的风险:如果AMD CI环境的配置与验证环境有差异,可能导致测试失败,但PR body已通过dispatched CI验证确认通过。

  • 用户:无直接影响。
  • 系统/CI:AMD夜间CI将新增5个JIT内核测试,增加约45-120秒的运行时间,确保AMD平台对这些内核的持续兼容性。
  • 团队:AMD相关开发者将获得更及时的回归反馈。
纯测试基础设施变更

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论