AMD与英特尔统一x86版PyTorch优化
IT时代网10月21日消息,AMD 与英特尔宣布展开合作,把双方面向 PyTorch 的 x86 处理器优化统一到开源深度学习库 oneDNN 中,为开发者提供统一的软件路径。
按照计划,AMD 将把自家的 ZenDNN 优化并入 oneDNN,同时继续针对自家处理器做优化。两家公司在 x86 生态顾问小组的建议下,已就 oneDNN 作为 PyTorch 中 x86 加速的默认开源库达成一致。oneDNN 由统一加速基金会管理。
AMD 架构与战略企业副总裁表示,公司正在推进开放的软件生态,通过 x86 生态顾问小组把行业聚集到通用技术之上,简化跨 x86 处理器的开发工作。
英特尔数据中心战略项目副总裁表示,开发者不需要为了在 x86 处理器上获得最佳表现而考虑不同的软件路径,与 AMD 通过 oneDNN 合作为 PyTorch 社区提供了一致的方案,也为后续把 AI 计算扩展等能力引入框架打下基础。
红帽首席技术官办公室的一位副总裁称,两家公司在 PyTorch 中通过 oneDNN 统一 x86 处理器优化,是消除生态碎片化的重要一步,会让开发者更容易在不同 x86 架构上一致地构建、优化与部署 AI 负载。
此外,双方还在 AI 计算扩展上展开合作。这套面向 x86 的新一代指令集架构用于加速矩阵乘法等 AI 核心运算,未来将通过 oneDNN 更新接入 PyTorch。AMD 同时以指导委员会成员身份加入统一加速基金会。

编辑:林一舟