IBM Spyre 开源 MLIR 数据流后端，KTIR 降为 SpyreCode

IBM Spyre 团队在 PyTorch 博客发布并开源了 Dataflow Scheduler。该调度器把 torch-spyre Inductor 生成的 KTIR 逐层降为 SpyreCode，而 KTIR 承载的核心划分与张量驻留决策由调度器继续沿用，后端无需重新推导这些映射。 团队称 Dataflow Scheduler 1.0 版已就绪，基于上游 LLVM/MLIR 构建。KTIR、调度器的 ScheduleIR 和 DataflowIR 都是基于 linalg、scf、arith、memref、tensor 等上游方言的普通 MLIR；调度器及其方言以 Apache-2.0 协议开源。 据团队介绍，该版本支持多核 KTIR 程序中带广播和归约维度的逐元素操作，并已用典型 transformer 操作在从 KTIR 经调度到代码生成的完整链路上做了端到端验证。

AI 根据报道生成 · 3 小时前更新