PyTorch Blog· Swagath Venkataramani, Bardia Mahjour, Prasanth Chatarasi, Alberto Mannari, Viji Srinivasan, and the IBM Spyre Team·原文发布· 3 小时前AI 评分38
从 KTIR 到 SpyreCode：面向 IBM Spyre 的 MLIR 数据流后端
From KTIR to SpyreCode: An MLIR Dataflow Backend for IBM Spyre
AI 导读
PyTorch 博客介绍 torch-spyre 后端编译链：Inductor 的切分决策通过 KTIR（Kernel Tile IR，基于 MLIR 的数据并行表示）传递给后端，KTIR 假设每个计算 tile 可见所有内存，单个逻辑张量可分布在多个核的 scratchpad 上。
本站提供 AI 导读，完整报道请前往来源网站阅读。
阅读原文 PyTorch Blog · pytorch.org