Printf直调样例说明
概述
本样例通过Ascend C编程语言实现了Matmul算子，使用<<<>>>内核调用符来完成算子核函数在NPU侧运行验证的基础流程，同时在算子中添加printf调测，给出了对应的端到端实现。
支持的产品
- Atlas A3 训练系列产品/Atlas A3 推理系列产品
- Atlas A2 训练系列产品/Atlas A2 推理系列产品
目录结构介绍
├── 00_printf
│ ├── scripts
│ │ ├── gen_data.py // 输入数据和真值数据生成脚本
│ │ └── verify_result.py // 验证输出数据和真值数据是否一致的验证脚本
│ ├── CMakeLists.txt // 编译工程文件
│ ├── data_utils.h // 数据读入写出函数
│ └── printf.asc // Ascend C算子实现 & 调用样例
算子描述
-
算子功能：
Matmul实现了快速的Matmul矩阵乘法的运算操作。
printf可以实现调试场景下的格式化输出功能。
本样例将打印的内容保存到输出文件中，并对比文件中是否有打印的内容，以及样例正常执行未中断报错，从而判断样例是否执行成功。
Matmul的计算公式为：
C = A * B + Bias- A、B为源操作数，A为左矩阵，形状为[M, K]；B为右矩阵，形状为[K, N]。
- C为目的操作数，存放矩阵乘结果的矩阵，形状为[M, N]。
- Bias为矩阵乘偏置，形状为[N]。对A*B结果矩阵的每一行都采用该Bias进行偏置。
在上述算子中添加printf，可以打印执行过程中的部分变量值，变量类型包括无符号整型、整型、字符型、bool类型、half和float浮点型等等，可以方便调试。
-
算子规格：
在核函数直调样例中，算子实现支持的shape为：M = 512, N = 1024, K = 512。算子类型(OpType) Matmul 算子输入 name shape data type format a M * K float16 ND b K * N float16 ND bias N float ND 算子输出 c M * N float ND 核函数名 matmul_custom -
算子实现：
本样例中实现的是[M, K, N]固定为[512, 512, 1024]的Matmul算子，并调用printf进行调测。-
kernel实现 Matmul算子的数学表达式为：
C=A∗BC = A * B
其中A的形状为[512, 512], B的形状为[512, 1024], C的形状为[512, 1024]。
-
调用实现 使用内核调用符<<<>>>调用核函数。
-
编译运行
-
配置环境变量
在本样例根目录下执行如下步骤，编译并执行算子。 请根据当前环境上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; # 编译工程 python3 ../scripts/gen_data.py # 生成测试输入数据 ./demo # 执行编译生成的可执行程序，执行样例 python3 ../scripts/verify_result.py output/output.bin output/golden.bin # 验证输出结果是否正确，确认算法逻辑正确最终执行结果如下，说明精度对比成功。
test pass!