IBM 开源面向 Spyre 的 KTIR 与 Dataflow Scheduler 编译器

IBM Spyre 团队发布技术博客并开源 KTIR（Kernel Tile IR）与 Dataflow Scheduler 编译器，用于 torch-spyre 后端将 PyTorch 程序编译到 IBM Spyre 加速器上。 按博客介绍，Inductor 的切分决策通过基于 MLIR 的 KTIR 传递给后端，KTIR 假设每个计算 tile 可见所有内存，单个逻辑张量可分布在多个核的 scratchpad 上。Dataflow Scheduler 第一版已就绪，可处理多核 KTIR 程序中带广播和归约维度的逐元素操作，并已在代表性 transformer 算子上完成从 KTIR 到调度和代码生成的端到端验证。

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

还没有足够的连续观测数据，暂不绘制趋势。