PyTorch 官方详解 IBM Spyre 后端：KTIR 与数据流调度器如何跑通 AI 加速器
PyTorch · x · 2026-10-21
PyTorch 官方博客发布 IBM Spyre 团队的深度技术文章，讲解 PyTorch 工作负载如何被翻译成 AI 加速器可执行程序。
- KTIR（Kernel Tile IR）：基于 MLIR 的中间表示，刻画数据并行的加速器视图——每个计算 tile 可见所有内存，单个逻辑张量可分布在多个核的 scratchpad 上
- Dataflow Scheduler：新开源组件，把 Inductor 决定的work分配、内存放置与 tiling 展开为嵌套、双缓冲、多实例的数据流流水线，输出 DataflowIR
- SpyreCode：最终产物为带符号张量地址的 SpyreBinary 加 job plan，运行时只准备一次、启动时填入具体地址与形状
这是 torch-spyre 三部曲系列的中间一章，代码已开源。
「Infra」频道最新
- Google 开源 TorchTPU，联手 Amazon 挖角 NVIDIA 的 PyTorch 生态 — gaganghotra_ · 2026-10-21
- FastVideo 开源 FastH3 Omni：Blackwell 上实时生成快 12 倍 — multimodalart · 2026-10-21
- Anthropic IPO 草案曝光：上半年营收 163 亿美元，净亏 800 亿 — kimmonismus · 2026-10-21
- dphnAI 分布式推理网络日供 320 亿 DeepSeek tokens,约为 Venice 日量 6% — Ronangmi · 2026-10-21
- 黄仁勋:未来每家公司都将拥有两座工厂,一座真实一座数字 — round · 2026-10-21
- DeepSeek V4 被评「神级 infra + 四流 ML」:一次发布堆 7 个技巧 20 个补丁 — teortaxesTex · 2026-10-21