viable/strict/1788417739: [ROCm] Update rocm preview version to 21st August wheel (#193181)
PyTorch 将 ROCm 预览包更新至 10.1.0a20260821,并在 Ubuntu ROCm 镜像中安装 liblzma-dev 以满足 AOTriton 0.13 的配置需求。同时,在 ROCm 10.1 及更高版本上跳过 TestCompileKernel 和 test_graph_external_wait_and_record 测试,因其在 HIPRTC 中失败,相关运行时问题(AIRUNTIME-2707)正在调查修复中。
Development
- First Reportviable/strict/1788417739: [ROCm] Update rocm preview version to 21st August wheel (#193181)PyTorch Core
- Industry Responsetrunk/1024f023624bf4bb58ea48df79c49f992d1e3c46: [ROCm] Update rocm preview version to 21st August wheel (#193181)PyTorch Core
- Current AssessmentPyTorch 对 ROCm 的持续支持反映了 AMD 在 AI 加速领域的地位提升。跳过测试和修复运行时问题表明 ROCm 软件栈仍在追赶 CUDA 的成熟度,但 PyTorch 的积极适配有助于 AMD 在 AI 训练和推理中获得更多采用。Agent Pulse · analysis
PyTorch 在 2026 年 9 月 3 日发布了针对 ROCm 的更新,将预览包版本提升至 10.1.0a20260821。该更新在 Ubuntu ROCm 镜像中安装了 liblzma-dev,因为 AOTriton 0.13 在配置阶段需要此包(通过 pkg_search_module(liblzma) 检测)。此外,由于 TestCompileKernel 和 test_graph_external_wait_and_record 在 HIPRTC 中失败,这些测试在 ROCm 10.1 及更高版本中被跳过。ROCm 运行时问题(AIRUNTIME-2707)正在调查和修复中。测试计划包括在 PR 上运行 ciflow/rocm-preview,确认预览构建完成以及 HIPRTC 测试在 ROCm 10.1+ 上被跳过。
ROCm 10.1 预览版的更新表明 PyTorch 正在积极适配 AMD GPU 的新版本。跳过 HIPRTC 相关测试暗示 ROCm 10.1 的 HIPRTC 组件存在兼容性问题,可能影响内核编译和图形同步功能。AOTriton 0.13 对 liblzma 的依赖表明 Triton 编译器在配置阶段需要压缩库支持,这可能是为了处理内核二进制或元数据的压缩。
PyTorch 对 ROCm 的持续支持反映了 AMD 在 AI 加速领域的地位提升。跳过测试和修复运行时问题表明 ROCm 软件栈仍在追赶 CUDA 的成熟度,但 PyTorch 的积极适配有助于 AMD 在 AI 训练和推理中获得更多采用。
对于使用 AMD GPU 进行 AI 开发的组织,此更新确保 PyTorch 与最新 ROCm 版本的兼容性,减少因版本不匹配导致的构建失败。跳过已知失败的测试可避免 CI 中断,提高开发效率。
随着 AIRUNTIME-2707 的修复,预计后续版本将重新启用这些测试,并可能进一步优化 ROCm 10.1 上的性能。PyTorch 与 ROCm 的集成将更加稳定,为 AMD GPU 用户提供更好的体验。