CANN训练营第三季-昇腾CANN算子精讲课-学习笔记

CANN训练营第三季-昇腾CANN算子精讲课-学习笔记

算子适配插件是用来实现第三方网络框架中算子到CANN算子映射的。

TBE TIK开发方式对数据的操作比较灵活，用户可以手工控制数据的搬运。

TBE DSL算子开发方式提供了计算接口与自动调度功能，更加简单便捷。

TBE DSL和TBE TIK开发的算子是跑在AI Core上的。

张量描述符包含如下属性：名称 形状 数据类型 数据排布格式。

一个网络中可以包含相同类型（Type）不同名称（Name）的算子。

UT测试类中，ElementwiseOpUT测试类可以用来测试单输入单输出类型算子。

编写TBE算子的DSL语言dsl.vlog(data)。

从算子原型库中可以获得下面的信息：算子类型 输入个数 输入数据类型。

TBE DSL算子代码实现时的必选的步骤：

导入依赖的Python模块

为输入Tensor占位

调用计算接口实现逻辑

调用自动调度接口auto_schedule

调用编译接口

TBE DSL算子的编译过程包括Schedule -> Pass -> Codegen。

ST测试包括下列步骤

创建测试用例定义文件

修改测试用例定义文件

执行测试用例

ST测试不会测试算子的适配插件。

ST测试前需要编译部署一次，后续修改算子实现代码需要再重新部署。