自定义算子开发样例

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

此样例包含了TBE自定义算子、AI CPU自定义算子以及TensorFlow Scope融合规则开发的代码样例，同时提供了算子编译规则文件。您可以直接基于此样例编译出自定义算子安装包，也可以基于此Sample追加自己的自定义算子实现代码，进行自定义算子的开发。

更多样例请在CANN软件安装后文件存储路径中的“tools/msopgen/template/custom_operator_sample”目录下获取，具体信息请参考自定义算子模板。

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

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

表1 TBE DSL样例算子列表
算子名称	样例描述
Add	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Add_CN.md”路径中获取详细的样例描述。
LeakyRelu	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/LeakyRelu_CN.md”路径中获取详细的样例描述。

表2 TBE TIK样例算子列表
算子名称	样例描述
Conv2d	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Conv2d_CN.md”路径中获取详细的样例描述。
Matmul	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Matmul_CN.md”路径中获取详细的样例描述。
Permute	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Permute_CN.md”路径中获取详细的样例描述。
ScatterNdAdd	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/ScatterNdAdd_CN.md”路径中获取详细的样例描述。
Upsample	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Upsample_CN.md”路径中获取详细的样例描述。

表3 AI CPU样例算子列表
算子名称	样例描述
addblockcust	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/addblockcust_CN.md”路径中获取详细的样例描述。
ReshapeCust	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/ReshapeCust_CN.md”路径中获取详细的样例描述。
UniqueCust	单击Gitee或Github，进入Ascend samples开源仓，参见README中的“版本说明”下载配套版本的sample包，从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/UniqueCust_CN.md”路径中获取详细的样例描述。

父主题： 样例参考