跳到正文
PyTorch Blog· Joshua Rosenkranz, Tuan Hoang Trong, Thomas Gooding, Matthew Pisano, and the IBM Spyre Team·原文发布· 7 天前AI 评分34

将 Spyre 构建为原生 PyTorch 设备

Building Spyre as a Native PyTorch Device

AI 导读

IBM 的 torch-spyre 团队通过 PrivateUse1 为 Spyre 数据流 AI 加速器注册设备身份,并接入 PyTorch 的分配器、流、事件与编译抽象。张量可常驻 device="spyre",流提供顺序语义,事件连接跨流依赖,FX 图仍走 Inductor 编译路径。该集成统一了 eager 与编译执行路径,降低启动开销。

本站提供 AI 导读,完整报道请前往来源网站阅读。

阅读原文 PyTorch Blog · pytorch.org