简介

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

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

在当前版本中，TorchAir仅作为一项试用特性，此功能在后续版本中可能会有所调整或改进。请用户在使用过程中关注后续版本的迭代。