AMD与英特尔合作统一PyTorch x86 CPU优化路径

10月20日,AMD与英特尔共同宣布,双方将围绕PyTorch在x86 CPU上的优化展开合作。按计划,两家公司会把各自现有的处理器优化工作集中到开源深度学习库oneDNN中,使其成为PyTorch内x86加速的默认开源库。AMD方面也会把ZenDNN优化集成到oneDNN,并加入统一加速基金会(UXL Foundation),担任指导委员会成员。

AMD与英特尔合作统一PyTorch x86 CPU优化路径

这一合作由x86生态咨询组推动。对开发者来说,过去可能需要针对AMD或英特尔CPU分别适配和调优PyTorch工作负载,之后则有望通过同一个软件路径获得优化支持。若推进顺利,开发者、云服务商和企业用户在x86服务器、工作站及AI PC上可以获得更一致的PyTorch性能体验。

除oneDNN之外,AMD与英特尔还在推进面向x86的下一代指令集架构“AI计算扩展”(ACE),重点加速矩阵乘法。矩阵乘法是AI模型训练与推理中的核心运算,相关能力未来计划通过oneDNN更新接入PyTorch。两家公司表示,这种统一路径有助于减少x86 AI生态的碎片化,降低重复开发与维护成本,改善AI应用在构建、部署和跨平台迁移环节的体验。