简介

TorchAir（Torch Ascend Intermediate Representation）是为torch_npu提供图模式能力的扩展库，支持用户使用PyTorch和torch_npu在昇腾设备上进行图模式的训练和推理。TorchAir对外提供昇腾设备的图模式编译后端，对接PyTorch的dynamo特性，将PyTorch的FX计算图转换为昇腾的GE计算图，并提供在昇腾设备上启动GE计算图编译和执行的能力。

TorchAir继承了dynamo的大部分特性，如动态shape图功能，在PyTorch的基础上又新增了如下功能：

• 日志功能
• graph dump功能
• dynamo export功能
• data dump功能
• graph fusion功能
• experiment性能提升功能