昇腾社区首页
中文
注册

调用样例

完成了算子适配开发后,即可实现C++ extensions的方式调用自定义算子。下文介绍注册算子开发过程以及算子适配开发过程。昇腾提供了add_custom和matmul_leakyrelu_custom两个kernel算子适配样例供开发者参考,具体可查看LINK

  1. 完成自定义算子工程创建、算子开发及编译部署流程,具体可参考CANN Ascend C算子开发指南
  2. 下载示例代码。
    #LINK下载样例代码
    git clone https://gitee.com/ascend/op-plugin.git
    cd op-plugin
    git checkout master
    # 进入代码目录
    cd examples/cpp_extension
  3. 完成算子适配,具体可参考适配开发
  4. 执行如下命令编译并安装wheel包。
    python setup.py bdist_wheel
    cd dist
    pip install *.whl
  5. 执行测试用例并查看验证结果。

    执行如下命令:

    cd test
    python test.py
    返回如下回显,表示执行成功。
    1
    2
    3
    Ran 2 tests
     
    OK