调用样例
完成了算子适配开发后,即可实现C++ extensions的方式调用自定义算子。下文介绍注册算子开发过程以及算子适配开发过程。昇腾提供了add_custom和matmul_leakyrelu_custom两个kernel算子适配样例供开发者参考,具体可查看LINK。
- 完成自定义算子工程创建、算子开发及编译部署流程,具体可参考《CANN Ascend C算子开发指南》。
- 下载示例代码。
- 完成算子适配,具体可参考适配开发。
- 执行如下命令编译并安装wheel包。
python setup.py bdist_wheel cd dist pip install *.whl
- 执行测试用例并查看验证结果。
执行如下命令:
cd test python test.py
返回如下回显,表示执行成功。1 2 3
Ran 2 tests OK
父主题: kernel直调算子适配