简介
基于
Ascend C
方式实现矢量算子的流程如下图
图1
所示。
图1
矢量算子实现流程
算子分析:分析算子的数学表达式、输入、输出以及计算逻辑的实现,明确需要调用的
Ascend C
接口。
核函数定义:定义
Ascend C
算子入口函数。
根据矢量编程范式实现算子类:完成核函数的内部实现。
下文以ElemWise(Add)算子为例,对上述步骤进行详细介绍。本样例中介绍的算子完整代码参见
实现样例
。
父主题:
矢量编程