昇腾社区首页
中文
注册

自定义算子开发

样例获取

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

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

您也可以基于此样例追加自己的自定义算子实现代码,进行自定义算子的开发。

样例描述

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

  • 针对Ascend EP场景,算子编译时build.sh文件中的“TOOLCHAIN_DIR”请配置为HCC编译器所在路径,例如:

    export TOOLCHAIN_DIR=${INSTALL_DIR}/toolkit/toolchain/hcc

  • 针对Ascend RC场景(例如:Atlas 200 DK),算子编译时build.sh文件中的“TOOLCHAIN_DIR”请配置为g++交叉编译器所在bin文件夹的上级目录,例如,交叉编译器存储路径为/usr/bin/aarch64-linux-gnu-g++,则TOOLCHAIN_DIR配置如下:

    export TOOLCHAIN_DIR=/usr