实现Converter（可选）

如果您希望使用max-autotune模式功能，例如SuperKernel等高阶能力，需要额外实现Ascend Converter（使用Ascend IR表达算子的计算逻辑）。

在Eager模式下，my_op调用Ascend C算子MyOp；而对应到Converter实现，调用Ascend IR MyOp。

• 在Ascend C算子工程编译时，除了生成aclnnXxx接口外，还会同步生成同名Ascend IR的注册代码。
• 接口介绍参见register_fx_node_ge_converter和custom_op。

通常不需要手动实现Converter，TorchAir会自动完成PyTorch算子到同名（大驼峰）Ascend IR的转换。例如本样例中的my_op算子，会自动转换为Ascend IR MyOp。

如果自动转换无法完成，TorchAir的编译报错信息会给出原因，原因一般如下：

• PyTorch算子的名字无法与Ascend IR名字通过大驼峰格式对应，例如my_op实际对应Ascend IR的名字为MyOp或OtherOp等。
• PyTorch算子与Ascend IR的输入输出顺序或数量不一致。
• PyTorch算子原型定义中存在Scalar类型入参。

您可以修改PyTorch算子原型使其满足条件，让TorchAir自动完成转换，或者手动实现Converter：

在third_party/torchair/torchair/python/torchair/_ge_concrete_graph/ge_converter/custom目录下，新建MyOp算子对应的my_op.py文件，添加如下代码实现Converter：

1
2
3
4
5
6
7

import torch
import torchair

# torch.ops.npu.my_op.default为自定义算子生成的Python函数签名，注意default后缀
@torchair.register_fx_node_ge_converter(torch.ops.npu.my_op.default)  
def convert_npu_my_op(x, y, z, attr1, attr2):           # 函数入参与Torch算子入参一致
    return torchair.ge.custom_op("MyOp", x, y, z, attr1, attr2)