PyTorch Blog· Joshua Rosenkranz, Tuan Hoang Trong, Thomas Gooding, Matthew Pisano, and the IBM Spyre Team·· 3 小时前AI 评分37
IBM 如何将 Spyre 打造为 PyTorch 原生设备
Building Spyre as a Native PyTorch Device
阅读原文
本站未展示全文，请前往来源网站阅读。
AI 导读
IBM 通过 torch-spyre 将 Spyre 接入 PyTorch 的 PrivateUse1 设备抽象，使张量可常驻 device="spyre"，并支持流、事件与 Inductor 编译路径。
来源：PyTorch Blog · pytorch.org