当前位置：首页 > news >正文

CANN/asc-devkit SIMD向量函数Dump接口

news 2026/5/30 21:59:46

SimdVF Dump接口功能说明

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

概述

本样例介绍simd_vf侧asc_dump系列接口使用方法，在vector函数中通过asc_dump_ubuf、asc_dump_reg、asc_dump等接口实现UB和寄存器数据的可视化打印。核函数从二进制文件读取输入数据，执行Adds计算后写出输出数据，通过比较输出与真值进行校验。

支持的产品

Ascend 950PR/Ascend 950DT

目录结构介绍

├── simd_vf_dump │ ├── scripts │ │ ├── gen_data.py // 输入数据和真值数据生成脚本 │ │ └── verify_result.py // 验证输出数据和真值数据是否一致 │ ├── CMakeLists.txt // 编译工程文件 │ ├── data_utils.h // 数据读入写出函数 │ └── simd_vf_dump.asc // Ascend C样例实现&调用样例

样例描述

样例功能：
使用vector编程模式，展示simd_vf侧asc_dump系列接口的基本使用方法。核函数通过ReadFile读取二进制输入数据，使用asc_vf_call调用simd_vf函数进行dump打印，执行Adds计算后通过WriteFile写出二进制输出数据，最终通过校验脚本比对输出与真值。
1. asc_dump_ubuf<T>-打印UB数据
2. asc_dump_reg<T>-打印寄存器数据
3. asc_dump<T>-打印UB/Reg数据（通用接口）

simd_vf dump接口用法：

接口	地址类型	示例
`asc_dump_ubuf<T>`	`__ubuf__`	`asc_dump_ubuf<float>(input, desc, dump_size);`
`asc_dump_reg<T>`	Reg	`asc_dump_reg<float>(input, desc, dump_size);`
`asc_dump<T>`	`__ubuf__`/ Reg	`asc_dump<float>(input, desc, dump_size);`

参数说明：

模板参数T：指定数据类型（float/half等）
input：UB数据地址或Reg寄存器对象
desc：打印控制描述符
dump_size：单次dump的数据元素个数

调用实现
在__global__ __vector__核函数内使用asc_vf_call<FuncName>()调用simd_vf函数完成dump打印，随后通过AscendC::Adds执行计算并将结果写回GM。Host侧通过ReadFile/WriteFile完成二进制数据的读写。

编译运行

配置环境变量在本样例根目录下执行如下步骤，编译并执行样例。请根据当前环境上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 -DCMAKE_ASC_ARCHITECTURES=dav-3510 ..;make -j; # 编译工程 python3 ../scripts/gen_data.py # 生成测试输入数据 ./demo # 执行编译生成的可执行程序，执行样例 python3 ../scripts/verify_result.py output/output.bin output/golden.bin # 验证输出结果是否正确，确认算法逻辑正确