快科技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中。
![]()
特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。
Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.