Prhub

#55681 [XPU][Tests] Enable test_per_token_group_quant_int8 on XPU

原始 PR 作者 pmanczak 合并时间 2026-09-11 15:59 文件变更 1 提交数 1 评论 3 代码增减 +5 / -2

执行摘要

XPU 放开 int8 分组量化测试,设备改为平台自适应

PR body 明确指出问题:test_per_token_group_quant_int8 was gated on torch.cuda.is_available() and hardcoded device="cuda", so it never ran on XPU even though int8_utils.per_token_group_quant_int8 already has a Triton fallback for any non-CUDA-alike device。commit message 进一步说明:The skipif guard and hardcoded device="cuda" left this test CUDA-only。也就是说被测算子在 XPU 上本就可执行(走 Triton 回退),是测试自身的平台判定把 XPU 挡在门外,属于测试与实现能力不匹配导致的白白丢失覆盖率。Test plan 给出的证据是在 Intel Arc Pro B70(XPU)上执行该用例得到 6 passed(此前为 6 skipped,原因是 CUDA not available)。

值得快速浏览而非精读:它是一个只有 5 行增量的测试解禁,但包含两个可复用的决策点。第一,平台守卫应表达「这个后端是否被验证过」而不是「本机是否有 CUDA」,用 current_platform.is_cuda_alike() or current_platform.is_xpu() 表达;第二,设备绑定统一为 current_platform.device_type,与同文件 test_per_token_group_quant_fp8 保持一致。若你正在为某一后端补测试,这个 PR 是很好的最小模板;若你关心测试有效性,需要注意在 XPU 上该用例实际只覆盖了 Triton 回退路径而非平台专用内核,并且它放弃了「无 GPU 即跳过」的保护。

讨论亮点

该 PR 的讨论非常轻量,没有出现技术争议:claude[bot] 两次给出同一句提示 —— This pull request is from a fork — automated review is disabled,意味着来自 fork 的变更没有自动静态审查,需要维护者手动 @claude review 才能触发一次性审查,而本次并未触发。coderabbitai[bot] 的总结是 No actionable comments were generated in the recent review,即未提出可执行问题。人工侧由 yma11 先给 LGTM 并点名 @jikunshang PTAL,jikunshang 随后 APPROVED,两位维护者均未留下行内评论(review_comments_count 为 0)。作者则通过 /ci run 触发 CI,github-actions[bot] 回执 Triggered Buildkite CI #87633 for commit d0a196ae4d9b。整体是「低风险测试解禁 + 双人快速批准」的合入路径。

实现拆解

  1. 变更入口与范围:唯一改动文件是 tests/kernels/quantization/test_per_token_group_quant.py,改动符号是 test_per_token_group_quant_int8,新增 5 行、删除 2 行,全部集中在装饰器与函数首行的设备绑定上,测试主体断言逻辑一行未动。
  2. 跳过条件从「设备可用性」改为「平台类型」:@pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA not available") 被替换为 @pytest.mark.skipif(not (current_platform.is_cuda_alike() or current_platform.is_xpu()), reason="Only test on CUDA/ROCm/XPU.")。语义上从「这台机器有没有 CUDA 设备」变成「这个构建的目标平台是不是 CUDA-alike 或 XPU」,从而让 ROCm 与 XPU 走到同一守卫下,同时仍排除纯 CPU 等未验证后端。
  3. 设备解析与同文件既有写法对齐:device = "cuda" 改为 device = current_platform.device_type;commit message 说明这是 to derive the device the same way test_per_token_group_quant_fp8 does,即复用同文件 test_per_token_group_quant_fp8 已经采用的平台自适应模式,避免一个文件里出现两种设备约定。
  4. 无新增 import 与依赖变化:本 PR 只有 +5/-2 共 7 行变更,恰好等于守卫块(4 行新增、1 行删除)加设备行(1 行新增、1 行删除),没有插入任何 import 语句,说明 current_platform 已在模块级导入并被同文件其他测试使用;因此不存在新增依赖。
  5. 测试矩阵与验证配套:参数化仍是 3 个 shape((32,128)、(64,256)、(16,512))× 2 个 group_size(64、128)= 6 个用例,与作者报告的 6 passed 完全对应;测试内部依旧通过 patch("vllm.platforms.current_platform.is_cuda_alike", return_value=False) 构造 Triton 参考路径。本次没有新增测试文件、没有 CI 配置改动、没有生产代码或 schema 变更,验证依赖 Buildkite CI #87633 与本地 XPU 实机结果。
文件 模块 状态 重要度
tests/kernels/quantization/test_per_token_group_quant.py 量化测试 modified 3.76

关键符号

test_per_token_group_quant_int8

关键源码片段

tests/kernels/quantization/test_per_token_group_quant.py test-coverage

唯一改动文件;`test_per_token_group_quant_int8` 的 skipif 守卫与设备绑定被改为平台自适应,使 XPU 首次可以执行该用例,是本次覆盖率增量的全部来源。

@pytest.mark.parametrize("shape", [(32, 128), (64, 256), (16, 512)])
@pytest.mark.parametrize("group_size", [64, 128])
# 变更前:@pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA not available"),
# 只要本机没有可用 CUDA 设备就整体跳过,XPU 因此从未执行过这条用例;
# 变更后:只在 CUDA-alike(CUDA/ROCm)或 XPU 平台执行,
# 其余后端(如纯 CPU)仍然跳过,避免误报失败。
@pytest.mark.skipif(
    not (current_platform.is_cuda_alike() or current_platform.is_xpu()),
    reason="Only test on CUDA/ROCm/XPU.",
)
def test_per_token_group_quant_int8(shape, group_size: int):
    # 关键改动:不再写死 "cuda",改为跟随当前平台(XPU 上即 "xpu"),
    # 与同文件 test_per_token_group_quant_fp8 的设备解析方式保持一致。
    device = current_platform.device_type
​
    torch.manual_seed(42) # 固定随机种子,保证被测路径与参考路径拿到同一份输入
    num_tokens, hidden_dim = shape
​
    # 乘以 8 放大数值范围,让 int8 量化的舍入误差更容易暴露出来
    x = torch.randn((num_tokens, hidden_dim), device=device, dtype=torch.bfloat16) * 8
​
    # 被测路径:CUDA 上走专用 int8 内核;XPU 上 is_cuda_alike() 为 False,
    # 因此这里落到 Triton 回退实现——这正是本 PR 希望在 XPU 上覆盖的代码
    out_q, scale = int8_utils.per_token_group_quant_int8(
        x,
        group_size,
    )
​
    # 参考路径:把 is_cuda_alike() patch 成 False,强制两条路径都使用 Triton 实现。
    # 在 CUDA 上它构成「专用内核 vs Triton」的交叉校验;
    # 在 XPU 上两次调用等价,退化为 Triton 自洽性检查,
    # 主要价值是验证该内核能在 XPU 上编译并跑通全部 shape/group_size 组合
    with patch("vllm.platforms.current_platform.is_cuda_alike", return_value=False):
        ref_q, ref_s = int8_utils.per_token_group_quant_int8(
            x,
            group_size,
        )
​
    # 近似比较而非位级比较:不同内核的舍入策略不同,容差沿用原有阈值
    assert torch.allclose(out_q.float(), ref_q.float(), atol=0.15, rtol=0.15)
    assert torch.allclose(scale, ref_s, atol=0.01, rtol=0.01)

评论区精华

合入前的人工审批链 other

yma11 给出 LGTM. @jikunshang PTAL.,jikunshang 随后 APPROVED;两位维护者都没有留下行内评论(review_comments_count 为 0),也没有提出任何技术异议。

结论:以「LGTM + 点名二次确认」的方式双人批准,未发现需要返工的问题。 · 已解决

fork 来源 PR 的自动审查被禁用 question

claude[bot] 两次给出同一提示:This pull request is from a fork — automated review is disabled,需维护者评论 @claude review 才能触发一次性审查,本次未触发;coderabbitai[bot] 则正常完成分析并给出 No actionable comments were generated in the recent review。

结论:该 PR 实际依赖 Buildkite CI 与两位维护者的人工判断,静态自动审查覆盖有限,未产生可执行评论。 · informational

CI 触发与验证记录 other

作者执行 /ci run,github-actions[bot] 回执 Triggered Buildkite CI #87633 for commit d0a196ae4d9b;配合 PR body 中 Intel Arc Pro B70 上 6 passed 的本地结果,构成本次唯一可核验的测试证据。

结论:合入前已完成一轮 CI 与一次 XPU 实机验证,但 CI 是否包含 XPU 内核用例未在讨论中明确。 · 已解决

风险与影响

  1. 守卫语义变化可能带来无 GPU 环境的误跑:旧守卫 torch.cuda.is_available() 在装有 CUDA 版 torch 但无可见 GPU 的机器上会跳过;新守卫只看平台类型,current_platform.is_cuda_alike() 在这类环境下可能仍为 True,函数随后会执行 torch.randn(..., device=current_platform.device_type) 并失败。这是把「设备可用」换成「平台类别」时最典型的行为差异,属于测试基础设施层面的回归风险。
  2. XPU 上的覆盖含金量有限:被测量路径在 XPU 上因为 is_cuda_alike() 为 False 实际走的就是 Triton 回退,而参考路径又用 patch(..., return_value=False) 强制走同一实现,两条路径在 XPU 上等价,断言退化为 Triton 自洽性 + 编译可跑通的冒烟验证,并不能证明 XPU 存在专用 int8 内核或做了平台特化优化。
  3. 近似断言可能掩盖系统性偏差:断言使用 torch.allclose(..., atol=0.15, rtol=0.15) 与 atol=0.01, rtol=0.01 的宽松容差,扩展平台后如果某后端的 Triton 实现在 group_size=128 等组合上有系统性误差,仍可能落进容差内而无法暴露。
  4. 依赖外部 XPU 队列的验证:作者只在 Intel Arc Pro B70 单卡上验证,XPU CI 覆盖是否长期稳定取决于 Buildkite XPU 节点;一旦 XPU 队列缺位,这条用例会重新变成「无人执行」状态而不被察觉。
  5. 变更集合内不含生产代码,因此没有算子正确性、性能或 API 兼容性风险,用户侧行为零变化。

对用户与线上服务:无任何影响,改动仅在 tests/ 目录下,不改变 int8_utils.per_token_group_quant_int8 的分派逻辑、默认精度或性能。对系统与 CI:XPU 流水线新增 6 个实际执行的量化用例,把此前长期被 skip 的空白补上,属于覆盖率净增;同时 CUDA/ROCm 上的执行条件由「有可用 CUDA 设备」变为「平台为 CUDA-alike」,对常规 GPU CI 无实质变化。对团队:确立了「测试设备应与平台解耦、用 current_platform.device_type + is_cuda_alike()/is_xpu() 判定」的写法范式,后续 Intel GPU 相关的内核测试可以照此解禁;但同时也提示维护者在改用平台判定时要额外思考无设备环境的跳过逻辑。

仅测试变更 平台谓词语义变化 XPU 上覆盖价值有限 依赖外部 XPU 验证

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论