torchair:基于 PyTorch 与昇腾 NPU 的图模式推理加速项目

用户可将 PyTorch 模型在昇腾 NPU 上以图模式高效运行,实现推理加速。该项目继承 Dynamo 模式,将 FX 图转换为 GE 计算图并提供编译执行能力,支持日志、图导出、精度调试及自定义算子扩展等功能。【此简介由AI生成】

分支36Tags0

项目介绍

用户可将 PyTorch 模型在昇腾 NPU 上以图模式高效运行,实现推理加速。该项目继承 Dynamo 模式,将 FX 图转换为 GE 计算图并提供编译执行能力,支持日志、图导出、精度调试及自定义算子扩展等功能。【此简介由AI生成】

定制我的领域