PyTorch Blog· Joshua Rosenkranz, Tuan Hoang Trong, Thomas Gooding, Matthew Pisano, and the IBM Spyre Team·· 3 小时前AI 评分44
IBM 将 Spyre 构建为原生 PyTorch 设备
Building Spyre as a Native PyTorch Device
AI 导读
IBM 的 torch-spyre 团队通过 PrivateUse1 注册、PyTorch allocator 存储、streams、events 与 Inductor 编译路径，把 Spyre 接入为原生 PyTorch 设备。
来源：PyTorch Blog · pytorch.org