自定义算子开发

样例获取

单击GiteeGithub,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op”目录中获取样例。

此样例包含了TBE自定义算子、AI CPU自定义算子以及TensorFlow Scope融合规则开发的代码样例,同时提供了算子编译规则文件。开发者可直接基于此样例编译出自定义算子安装包。

您也可以基于此Sample追加自己的自定义算子实现代码,进行自定义算子的开发。更多样例请参考CANN安装路径中的“ascend-toolkit/latest/tools/msopgen/template/custom_operator_sample”目录下的算子。

您可以参见自定义算子模板,获取更多的自定义算子样例。

样例描述

您可以参见样例工程中的README.md详细了解每一个样例算子,并对样例算子工程进行编译、部署。

不建议对样例工程或自动生成的编译配置文件进行修改,否则可能会造成自定义算子运行失败。