英特尔宣布携手AMD,将双方针对PyTorch的x86 CPU优化成果,整合到开源深度学习库oneDNN中。这次合作将为PyTorch社区提供一条适用于现代处理器的统一路径,简化了跨x86 CPU的开发流程,无需为了充分发挥x86 CPU的性能而采用不同的软件路径。
打开网易新闻 查看精彩图片
oneDNN由统一加速基金会(UXL)管理,旨在提升AI工作负载在CPU上的运行效率。这一合作将为开发者提供一条通用软件路径,优化英特尔和AMD CPU上的AI工作负载,减少x86 AI生态的软件碎片化,进一步简化PyTorch应用在x86系统上的开发、维护和部署。
目前英特尔和AMD已就oneDNN达成一致,将其认定为PyTorch内x86加速的默认开源库。AMD正将ZenDNN优化整合进oneDNN,并以指导委员会成员的身份加入UXL基金会,通过开放协作,推进这项工作的进一步开展。
未来开发者可以通过一个统一的入口,获得面向英特尔和AMD处理器的PyTorch CPU优化,从而在不同的x86平台上获得更加一致的开发体验。英特尔和AMD围绕oneDNN的共同投入,为现有及未来的x86 AI能力提供了更为完善的支持。
双方还打算在AI计算扩展(AI Compute Extensions)方面展开合作,通过oneDNN更新,未来进一步集成至PyTorch中。这是面向x86设计的下一代指令集架构,能够加速矩阵乘法。
热门跟贴