PR body 指出该变更修复了使用 ROCm 7.13 构建 vllm 时的编译错误。ROCm 7.13 通过 PR rocm-systems#5194 在 HIP 中新增了 half 的 atomicAdd，与 vllm 已存在的定义冲突。
值得一读，展示了跨 GPU 平台 API 兼容性的处理方式，以及如何通过预处理版本检查优雅地适配上游变动。
gemini-code-assist 审核机器人指出直接定义 ROCM_VERSION_NUM 宏会导致全局命名空间污染，建议直接使用版本表达式。作者采纳建议，修改为内联版本检查。
csrc/quantization/gptq/compat.cuh 中定位到 atomicAdd_half2 和 atomicAdd(half) 的条件编译守卫。#if defined(__CUDA_ARCH__) || defined(USE_ROCM) 扩展为同时检查 ROCm 版本：仅当 (HIP_VERSION_MAJOR * 100 + HIP_VERSION_MINOR) < 713 时才启用自定义定义。ROCM_VERSION_NUM），直接在预处理表达式中内联版本计算，经 review 确认后采用此方式，防止宏污染。| 文件 | 模块 | 状态 | 重要度 |
|---|---|---|---|
csrc/quantization/gptq/compat.cuh |
量化内核 | modified | 3.7 |
csrc/quantization/gptq/compat.cuh
core-logic
核心文件，修改了 atomicAdd 条件定义中的版本守卫
// csrc/quantization/gptq/compat.cuh ( 关键片段 )
// 原有条件：无版本检查，导致 ROCm 7.13+ 上重复定义
// 修改后：仅当 ROCm 版本低于 7.13 时才定义自定义 atomicAdd
#if defined(__CUDA_ARCH__) || \
(defined(USE_ROCM) && (HIP_VERSION_MAJOR * 100 + HIP_VERSION_MINOR) < 713)
#if __CUDA_ARCH__ < 700 || defined(USE_ROCM)
__device__ __forceinline__ void atomicAdd(half* address, half val) {
// 自定义 half atomicAdd 实现
// ...
}
#endif
#endif
gemini-code-assist 建议不要定义 `ROCM_VERSION_NUM` 宏，而是直接在 `#if` 中使用 `HIP_VERSION_MAJOR * 100 + HIP_VERSION_MINOR`，以避免宏泄露到包含此头文件的任何文件中。
结论：作者采纳建议，删除了宏定义，改为内联版本表达式。 · 已解决
主要风险是兼容性：对于 ROCm 7.13 但尚未包含内置 atomicAdd 的中间版本，自定义定义被跳过可能导致链接错误。但根据 PR body 和评论讨论，该影响仅限于极窄的夜间构建窗口，且用户可通过更新到最新 ROCm 7.13 夜间版或回退到 7.12 解决。
仅影响 ROCm 平台上的量化内核编译（csrc/quantization/gptq/compat.cuh）。对 CUDA 和其他平台无影响。修复了 ROCm 7.13 用户的构建阻塞，影响范围小但关键。
当前没有检测到明确关联的 Issue 链接，后续同步到相关引用后会出现在这里。
参与讨论