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