执行摘要
该PR修复了ROCm CI和Docker构建因NVIDIA包命名变更（引入cu12/cu13后缀及无后缀变体）而失败的问题，通过扩展依赖排除列表和调整Dockerfile实现，但移除了PyTorch ROCm验证步骤，可能引入环境正确性风险。
功能与动机
PR body明确指出：“Fixes ROCm CI/Docker builds broken by nvidia package naming changes (cu12/cu13 suffixed and unsuffixed variants)。” 动机是解决NVIDIA包命名策略更新导致的构建依赖冲突，确保ROCm相关构建流程稳定运行。
实现拆解
实现涉及三个文件：
-
.pre-commit-config.yaml：在uv pip compile步骤中扩展--no-emit-package排除列表，新增三类NVIDIA包变体：
- 无后缀（如
nvidia-cublas）
-cu12后缀（如nvidia-cublas-cu12）-cu13后缀（如nvidia-cublas-cu13）
覆盖了cublas、cudnn、nccl等关键包，防止uv发出这些包导致依赖冲突。
-
requirements/rocm-test.txt：同步更新排除列表，确保与.pre-commit-config.yaml一致，例如将--no-emit-package, nvidia-cudnn-cu13改为--no-emit-package, nvidia-cudnn并添加所有变体。
docker/Dockerfile.rocm：将原有的PyTorch ROCm验证步骤（检查torch.version.hip）替换为持久化构建的wheel：
COPY --from=export_vllm /*.whl /opt/vllm-wheels/
目的是让后续脚本python_only_compile_rocm.sh能在移除编译器后重新安装wheel。
评论区精华
review中仅有的讨论聚焦于Dockerfile修改：
- gemini-code-assist[bot] 指出：
“While persisting the built wheel is a good addition, the verification step that ensures the PyTorch build is for ROCm has been removed. This check is a valuable safeguard to prevent building a Docker image with an incorrect PyTorch version.”
- tjtanaa 回复：
“Let's keep the original check”
但PR最终被tjtanaa批准合并，未在提交中重新添加验证步骤，此疑虑处于未解决状态。
风险与影响
- 主要风险：移除PyTorch ROCm验证步骤可能导致构建的Docker镜像错误使用CUDA版本PyTorch，影响ROCm环境正确性，可能引发运行时问题。
- 次要风险：大量手动排除项增加了维护负担，未来NVIDIA包命名再变更时需同步更新。
- 影响范围：仅限于CI和Docker构建流程，对最终用户无直接影响，但确保ROCm测试和部署的稳定性。
关联脉络
从近期历史PR看，该PR与以下PR同属基础设施修复范畴：
- #39390：修复CI夜间索引生成权限问题，同属CI脚本调整。
- #38950：在Dockerfile中添加fastsafetensors包，优化构建过程。
- #39164：为XPU跳过测试设置，解决CI死锁，同属平台特定CI调整。
整体趋势显示团队持续优化多平台（ROCm、XPU）的构建和测试基础设施，以应对硬件和依赖生态变化。
参与讨论