PyTorch Blog· Swagath Venkataramani, Bardia Mahjour, Prasanth Chatarasi, Alberto Mannari, Viji Srinivasan, and the IBM Spyre Team·· 4 小时前AI 评分39
从 KTIR 到 SpyreCode：IBM Spyre 的 MLIR 数据流后端
From KTIR to SpyreCode: An MLIR Dataflow Backend for IBM Spyre
阅读原文
本站未展示全文，请前往来源网站阅读。
AI 导读
IBM Spyre 的 torch-spyre 后端编译器通过 KTIR（Kernel Tile IR）承接 Inductor 已确定的并行划分，再由开源 Dataflow Scheduler 将其转换为数据流流水线，输出 DataflowIR 并最终生成 SpyreCode。
来源：PyTorch Blog · pytorch.org