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