跳到正文
原文
PyTorch· Dave Grove, Olivier Tardieu, IBM Spyre Team·· 9 小时前AI 评分42

PyTorch 为 IBM Spyre 加速器扩展 Inductor 后端

Extending Inductor for the IBM Spyre Accelerator

AI 导读

PyTorch 将 IBM Spyre 加速器的 torch.compile 后端从 sendnn 迁移至开源的 backend=inductor,新编译器体积缩小约 7 倍。

来源:PyTorch · pytorch.org