AMD与英特尔合作推进PyTorch的x86 CPU优化,选定oneDNN为共同路径

作者: 小蝶
更新时间: 2026-10-21 11:11

2026年10月20日,AMD与英特尔宣布合作,计划将双方针对PyTorch的x86 CPU优化工作整合到oneDNN。两家公司已选定oneDNN作为PyTorch中用于x86加速的默认开源库,希望让开发者通过共同的软件路径使用AMD和英特尔处理器上的AI优化。

AMD与英特尔合作推进PyTorch的x86 CPU优化,选定oneDNN为共同路径

oneDNN是一套由Unified Acceleration Foundation(UXL)管理的开源深度学习库,用于提高CPU执行AI工作负载的效率。这次合作由x86生态系统咨询小组建议推动,重点是减少两家处理器平台之间的软件优化分化。

AMD正在将其ZenDNN优化整合进oneDNN,并已以指导委员会成员身份加入UXL基金会。这里的进展是共同路径已经选定、优化整合正在进行,不能据此理解为所有工作都已进入现有PyTorch稳定版本。

对使用PyTorch开发AI应用的团队而言,双方希望减少为不同x86 CPU分别维护优化路径的负担,让应用构建、维护与部署更一致。实际性能仍取决于具体处理器、工作负载和软件版本;这次公告没有给出统一的性能提升幅度,也没有列出对应的PyTorch或oneDNN发布版本号。

合作还涉及AI Compute Extensions(ACE)。这是双方共同推进的下一代x86指令集架构扩展,主要面向AI计算中的矩阵乘法,计划通过未来的oneDNN更新接入PyTorch。公告将ACE接入列为未来更新安排,没有给出具体接入版本和时间。