样例运行验证
开发者调用Ascend C API实现自定义算子后,可通过单算子调用的方式验证算子功能。本代码仓提供部分算子实现及其调用样例,具体如下。
算子开发样例
| 目录名称 | 功能描述 |
|---|---|
| 01_simd_cpp_api | 基于Ascend C的SIMD API样例,通过<<<>>>直调的实现方式,介绍了SIMD API的使用方法 |
| 02_simd_c_api | 基于Ascend C的C_API样例,通过C_API的实现方式,介绍了C_API的使用方法 |
| 03_simt_api | 基于Ascend C SIMT编程的算子样例,通过<<<>>>直调的实现方式,介绍了SIMT的使用方法 |
npu-arch编译选项说明
开发者需根据实际的执行环境,修改具体样例目录下CMakeLists.txt文件中的--npu-arch编译选项,参考下表中的对应关系,修改为环境对应的npu-arch参数值。
| 产品型号 | npu-arch参数 |
|---|---|
| Ascend 950PR/Ascend 950DT | --npu-arch=dav-3510 |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
--npu-arch=dav-2201 |
| Atlas 推理系列产品AI Core | --npu-arch=dav-2002 |