viable/strict/1788197050: Move partial XPU device property utils to aten (#195146)
PyTorch 发布 viable/strict/1788197050 版本,将部分 XPU 设备属性工具从 intel/torch-xpu-ops 的 DeviceProperties.h 迁移到 aten,涉及 158 处 API 调用,如 syclMaxWorkGroupSize 改为 getKernelMaxWorkGroupSize,syclGpuEuCount 改用 getDeviceHWThreads 等。
发展脉络
- 首次出现viable/strict/1788197050: Move partial XPU device property utils to aten (#195146)PyTorch Core
- 当前判断PyTorch 将 XPU 设备属性工具迁移到 aten,表明英特尔与 PyTorch 在 XPU 支持上的合作加深,可能推动 XPU 在 AI 训练和推理中的采用。此举也可能简化第三方库对 XPU 的支持,降低开发门槛。Agent Pulse · 分析
PyTorch 在 2026 年 8 月 31 日发布版本 viable/strict/1788197050,将部分 XPU 设备属性工具从 intel/torch-xpu-ops 仓库的 DeviceProperties.h 迁移到 aten。该迁移旨在修复 intel/torch-xpu-ops 的 issue #4958,并将设备属性相关代码集中到一个文件中。迁移涉及 158 处 API 调用,例如 syclMaxWorkGroupSize 改为 getKernelMaxWorkGroupSize,syclMaxWorkItemsPerSubSlice 改为 getDeviceMaxWorkItemsPerXeCore,syclGpuEuCount 改用 getDeviceHWThreads 等。部分 API 被移除或替换,如 syclNativeVectorWidth 和 syclMaxNumSubGroups 被移除,syclGpuEuSimdWidth 被移除。
此次迁移将 XPU 设备属性查询从 torch-xpu-ops 的私有头文件提升到 aten 层,统一了设备属性 API 命名,并可能为后续 XPU 支持提供更稳定的接口。新 API 命名更清晰,如 getDeviceMaxWorkItems 替代 syclMaxWorkItemsPerTile,可能减少歧义。但迁移可能影响依赖旧 API 的代码,需要适配。
PyTorch 将 XPU 设备属性工具迁移到 aten,表明英特尔与 PyTorch 在 XPU 支持上的合作加深,可能推动 XPU 在 AI 训练和推理中的采用。此举也可能简化第三方库对 XPU 的支持,降低开发门槛。
对于使用 PyTorch 和 XPU 的企业,此次迁移可能简化部署和优化,降低对特定硬件厂商库的依赖。同时,统一的 API 可能减少维护成本,提升开发效率。
未来,PyTorch 可能继续将更多 XPU 相关代码迁移到核心库,进一步统一设备抽象。可关注后续版本是否移除旧 API,以及是否有更多 XPU 功能集成到 aten。