Add算子直调样例
概述
本样例演示基于静态Tensor编程模型的Add算子实现，该实现采用TQue内存管理机制实现数据搬运与计算任务的协同调度，特别适用于需要精细控制内存流水线的场景。
支持的产品
- Atlas A3 训练系列产品/Atlas A3 推理系列产品
- Atlas A2 训练系列产品/Atlas A2 推理系列产品
目录结构介绍
├── basic_api_tque_add
│ ├── CMakeLists.txt // 编译工程文件
│ └── add.asc // Ascend C算子实现,tque管理内存 & 调用样例
算子描述
-
算子功能：
Add算子实现了两个数据相加，返回相加结果的功能。对应的数学表达式为：
z = x + y -
算子规格：
算子类型(OpType) Add 算子输入 name shape data type format x 8 * 2048 float ND y 8 * 2048 float ND 算子输出 z 8 * 2048 float ND 核函数名 add_custom -
算子实现：
-
Kernel实现
Add算子的数学表达式为： 计算逻辑是：Ascend C提供的矢量计算接口的操作元素都为LocalTensor，输入数据需要先搬运进片上存储，然后使用计算接口完成两个输入参数相加，得到最终结果，再搬出到外部存储上。Add算子的实现流程分为3个基本任务：CopyIn，Compute，CopyOut。CopyIn任务负责将Global Memory上的输入Tensor xGm和yGm搬运到Local Memory，分别存储在xLocal、yLocal，Compute任务负责对xLocal、yLocal执行加法操作，计算结果存储在zLocal中，CopyOut任务负责将输出数据从zLocal搬运至Global Memory上的输出Tensor zGm中。
使用tque管理内存，使用静态Tensor编程方法进行Add算子的编程。
-
Tiling实现
TilingData参数设计，TilingData参数本质上是和并行数据切分相关的参数，其中basic_api_tque_add算子使用了2个tiling参数。使用的tiling参数为totalLength、tileNum。totalLength是指需要计算的数据量大小，tileNum是指每个核上总计算数据分块个数。比如，totalLength这个参数传递到kernel侧后，可以通过除以参与计算的核数，得到每个核上的计算量，这样就完成了多核数据的切分。
-
调用实现
使用内核调用符<<<>>>调用核函数。
-
编译运行
在本样例根目录下执行如下步骤，编译并执行算子。
- 配置环境变量
请根据当前环境上CANN开发套件包的安装方式，选择对应配置环境变量的命令。-
默认路径，root用户安装CANN软件包
source /usr/local/Ascend/cann/set_env.sh -
默认路径，非root用户安装CANN软件包
source $HOME/Ascend/cann/set_env.sh -
指定路径install_path，安装CANN软件包
source ${install_path}/cann/set_env.sh
-
- 样例执行
mkdir -p build && cd build; # 创建并进入build目录 cmake ..;make -j; # 编译工程 # 在build目录执行以下内容 ./add # 执行样例[Success] Case accuracy is verification passed.