跳到正文
PyTorch Blog· Swagath Venkataramani, Bardia Mahjour, Prasanth Chatarasi, Alberto Mannari, Viji Srinivasan, and the IBM Spyre Team·· 4 小时前精选AI 评分60

IBM Spyre 团队开源 Dataflow Scheduler:从 KTIR 到 SpyreCode 的 MLIR 数据流后端

From KTIR to SpyreCode: An MLIR Dataflow Backend for IBM Spyre

AI 导读

IBM Spyre 团队在 PyTorch 博客介绍 torch-spyre 后端编译链路,开源了 Dataflow Scheduler,把 Inductor 产出的 KTIR 经 ScheduleIR 降到 DataflowIR,再由设备相关阶段生成 SpyreCode。

推荐理由

IBM Spyre 团队公开了从 KTIR 到 SpyreCode 的完整编译链路,架构描述驱动的调度器设计对做数据流加速器编译的人有参考价值。

来源:PyTorch Blog · pytorch.org