返回资讯
开发与工具

将 Spyre 构建为 PyTorch 原生设备

Building Spyre as a Native PyTorch Device

来源配图图片仅供信息参考图片仅供参考版权归原作者
摘要

摘要:通过 torch-spyre,将 PyTorch 现有的设备、内存分配器、流和编译器抽象连接到 Spyre 运行时与固件,使 Spyre 成为 PyTorch 原生设备。PrivateUse1 为 Spyre 提供了真正的……本文最初发布于 PyTorch。

原始资料

PyTorch Blog · 来源摘要 · 正文节选

查看来源原文 产品参数、测试条件及最新修订以来源原文为准。