概念torch 扩展 / Torch Extension
Torch Extension
PyTorch 提供的 C++/CUDA 接口,允许厂商在不修改 PyTorch 主干代码的前提下,为自定义加速器实现算子、设备管理和内存分配逻辑,并以独立 Python 包的形式分发;常与 PrivateUse1 设备类型配合使用
时间轴 (近 90 天)
10月12日
torch扩展是PyTorch提供的一套C++/CUDA接口,允许厂商在不修改PyTorch主干代码的前提下,为自定义加速器实现算子、设备管理和内存分配逻辑,并以独立Python包的形式分发
待验证50%