快科技 10 月 21 日消息,Intel 与 AMD 联合宣布合作,将双方针对 PyTorch 的 x86 CPU 优化成果整合到开源深度学习库 oneDNN 中,减少 x86 AI 生态软件碎片化,简化 PyTorch 在 x86 系统上的开发部署。
oneDNN 由统一加速基金会(UXL)管理,旨在提升 AI 工作负载在 CPU 上的运行效率。
根据 x86 生态系统顾问小组的建议,Intel 和 AMD 已将 oneDNN 认定为 PyTorch 内 x86 加速的默认开源库。AMD 正将自家的 ZenDNN 优化整合进 oneDNN,并以指导委员会成员身份加入 UXL 基金会。
此前 Intel 有 oneDNN,AMD 有 ZenDNN,两套独立优化路径迫使开发者针对不同 CPU 分别适配,增加了维护成本。
之后开发者通过 oneDNN 这个统一入口,即可获得面向 Intel 和 AMD 处理器的 PyTorch CPU 优化,在不同 x86 平台上获得更一致的开发体验。
双方还在 AI 计算扩展(AI Compute Extensions)方面展开合作,这是面向 x86 设计的下一代指令集架构,能够加速矩阵乘法,未来将通过 oneDNN 更新进一步集成至 PyTorch 中。



登录后才可以发布评论哦
打开小程序可以发布评论哦