Prhub

#28967 [AMD] Register 7 JIT kernel unit tests for AMD nightly CI

原始 PR 作者 michaelzhang-ai 合并时间 2026-06-25 16:51 文件变更 7 提交数 2 评论 3 代码增减 +14 / -7

执行摘要

为 7 个 JIT 内核测试增加 AMD 夜间 CI 注册

在AMD GPU(MI35x / gfx950)上持续验证JIT内核的正确性,防止退化。PR body明确指出这些测试在ROCm 7.0和7.2上均通过,并提供了详细的运行时数据表格和CI任务链接。

该PR是标准的测试基础设施扩展,无技术复杂性,不值得精读。但可作为参考案例:如何在保持CUDA注册的同时,为AMD平台添加并行测试覆盖,且不破坏现有CI。

讨论亮点

无实质性review讨论。审核人HaiShaw直接批准,无任何评论。仅有amd-bot的CI状态评论确认所有CI失败与本PR无关。

实现拆解

  1. 导入调整:在每个目标测试文件中,将 from sglang.test.ci.ci_register import register_cuda_ci 改为 from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
  2. 新增注册行:在现有的 register_cuda_ci(...) 行之后,添加一行:register_amd_ci(est_time=<具体预估时间>, suite="nightly-amd-kernel-1-gpu", nightly=True)
  3. 验证:通过 run_suite.py --hw amd --suite nightly-amd-kernel-1-gpu --nightly 在ROCm 7.0和7.2上实际运行整个套件,所有7个测试均绿色通过。
文件 模块 状态 重要度
test/registered/jit/minimax/test_minimax_decode_topk.py 测试注册 modified 3.52
test/registered/jit/minimax/test_minimax_store_kv_index.py 测试注册 modified 3.52
test/registered/jit/test_deepseek_v4_compress_state_runtime_shapes.py 测试注册 modified 3.52
test/registered/jit/test_fused_store_index_cache.py 测试注册 modified 3.52
test/registered/jit/test_fused_verify_triton_gdn.py 测试注册 modified 3.52
test/registered/jit/test_kvcacheio_asymmetric.py 测试注册 modified 3.52
test/registered/jit/test_mla_kv_pack_quantize_fp8.py 测试注册 modified 3.52

关键源码片段

test/registered/jit/minimax/test_minimax_decode_topk.py test-coverage

新增 `register_amd_ci` 导入和注册行,将 MiniMax-M3 decode topk 测试纳入 AMD 夜间 CI。

# 变更仅涉及导入和注册行,其余代码不变
from sglang.jit_kernel.minimax_decode_topk import minimax_decode_topk
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci# 原有 CUDA 注册保持不变
register_cuda_ci(est_time=40, suite="base-b-kernel-unit-1-gpu-large")
register_cuda_ci(est_time=40, suite="base-b-kernel-unit-1-gpu-b200")
# 新增 AMD 夜间注册,est_time 基于实际运行耗时
register_amd_ci(est_time=15, suite="nightly-amd-kernel-1-gpu", nightly=True)

评论区精华

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

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

风险与影响

风险极低。变更纯粹是测试注册的添加,不涉及任何生产代码、工作流或依赖。唯一风险是这些测试在AMD环境下可能存在后续回归,但已被夜间CI覆盖,且PR已手动验证通过。

影响范围:仅AMD夜间CI。影响程度:低。7个JIT内核测试将自动在AMD nightly流水线中运行,有助于及早发现AMD平台上的内核退化。对CUDA、MUSA等其他平台无影响。

仅测试注册变更 无生产代码影响

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论