算子开发流程

cd \Documents
vim AddCustom.json

向AddCustom.json中添加如下内容，具体参考官方文档

[
    {
        "op": "AddCustom",
        "language": "cpp",
        "input_desc": [
            {
                "name": "x",
                "param_type": "required",
                "format": [
                    "ND"
                ],
                "type": [
                    "float16"
                ]
            },
            {
                "name": "y",
                "param_type": "required",
                "format": [
                    "ND"
                ],
                "type": [
                    "float16"
                ]
            }
        ],
        "output_desc": [
            {
                "name": "z",
                "param_type": "required",
                "format": [
                    "ND"
                ],
                "type": [
                    "float16"
                ]
            }
        ]
    }
]

使用msopgen工具创建Ascend C算子工程，参数详见官方文档

/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/bin/msopgen gen -i AddCustom.json -f tf -c ai_core-ascend310b -lan cpp -out ~/AddCustom

修改Host和Kernel侧的代码，这里作为样例，直接copy实例代码

cp ~/samples/operator/AddCustomSample/FrameworkLaunch/AddCustom/op_host/* ~/AddCustom/op_host/
cp ~/samples/operator/AddCustomSample/FrameworkLaunch/AddCustom/op_kernel/* ~/AddCustom/op_kernel/

修改算子工程中配置文件CMakePresets.json，将CMakePresets.json中ASCEND_CANN_PACKAGE_PATH的value值修改为实际CANN包安装目录/usr/local/Ascend/ascend-toolkit/latest

cd ~/AddCustom
vim CMakePresets.json

在AddCustom目录下执行如下命令编译自定义AddCustom算子，生成算子安装包。

./build.sh

部署自定义算子包

./build_out/custom_opp_ubuntu_aarch64.run

通过应用程序样例调用算子的ACLNN接口进行测试

cd ~/samples/operator/AddCustomSample/FrameworkLaunch/AclNNInvocation
bash run.sh

json 文件配置参数说明

详见官方文档

AI core 架构说明

详见官方文档