更新时间:2021-03-18 GMT+08:00
验证思路
自定义算子部署到算子库(OPP)后,可通过ATC(Ascend Tensor Compiler)工具将自定义算子转换成单个算子的离线模型文件(*.om),然后开发者可使用ACL(Ascend Computing Language)提供的单算子模型加载接口加载离线模型,并传入算子输入数据,进行算子执行,通过查看输出结果验证算子功能是否正确。
图1 自定义算子网络运行验证流程


- 将开发好的自定义算子部署到opp算子库对应目录的custom路径下,包含算子实现文件(*.py)、算子插件库(*.so)、算子原型库(*.so)、算子信息库(*.json)。
- 配置自定义算子json文件。
参照原型定义进行自定义算子的json文件配置,包括算子类型、属性、输入与输出。
- 使用ATC工具,加载2中的单算子json文件,生成只包含自定义算子的离线模型文件。
- 构造算子输入数据,用于进行算子功能验证。
- 进行ACL单算子验证的代码实现。
父主题: 网络测试用例实现
