PyTorch Blog· Dave Grove, Olivier Tardieu, IBM Spyre Team·· 21 小时前精选AI 评分67
PyTorch Inductor 如何扩展以支持 IBM Spyre 加速器
Extending Inductor for the IBM Spyre Accelerator
AI 导读
IBM 的 torch-spyre 将 `torch.compile` 后端从独立编译栈 sendnn 转向基于 PyTorch Inductor 扩展的 `backend=inductor`，替代编译器的代码量约减少 7 倍。
推荐理由
Spyre 后端的实现展示了 Inductor 如何适配 stick 数据布局、静态核心分工和本地 scratchpad，也呈现了这些约束对编译器扩展设计的影响。
来源：PyTorch Blog · pytorch.org