跳到正文
PyTorch:Blog· PyTorch:Blog(RSS)·· 3 小时前AI 评分22

IBM Spyre 加速器的 torch.compile 后端从 sendnn 迁移到 Inductor

AI 导读

IBM 为 Spyre 加速器打造的 PyTorch 软件栈已将 torch.compile 后端从自研的 sendnn 换成基于 PyTorch Inductor 扩展的 backend=inductor,编译器体积约为原来的 1/7。

来源:PyTorch:Blog · pytorch.org