文件最后提交记录最后更新时间
2 个月前
2 个月前
1 个月前
README.md

Acosh样例

概述

本样例基于Acosh高阶API计算反双曲余弦函数。

涉及样例: 除本样例使用的 Acosh 接口外,Ascend C 还提供了以下三角函数相关高阶API接口,除sincos外实现方式基本一致,如需调用替换接口名即可:

  • acos:反余弦函数。
  • asin:反正弦函数。
  • asinh:反双曲正弦函数。
  • atanh:反双曲正切函数。
  • cos:余弦函数。
  • cosh:双曲余弦函数。
  • sinh:双曲正弦函数。
  • tan:正切函数。
  • sincos:正弦余弦函数,分别计算正弦和余弦,调用时需要两个输出Tensor。

支持的产品

  • Ascend 950PR/Ascend 950DT
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品

目录结构介绍

├── acosh
│   ├── CMakeLists.txt     // 编译工程文件
│   └── acosh.asc          // Ascend C样例实现 & 调用样例

样例描述

  • 样例功能: 按元素做双曲反余弦函数计算,计算公式如下:

    dstTensori=Acosh(srcTensori)dstTensor_i = Acosh(srcTensor_i)

    Acosh(x)={Nan,x<1ln⁡(x+x2−1),x>1Acosh(x)=\begin{cases}Nan, & x < 1 \\ \ln(x+\sqrt{x^{2}-1}), & x > 1\end{cases}

  • 样例规格:

    样例类型(OpType) acosh
    样例输入
    nameshapedata typeformat
    src[1, 16]floatND
    样例输出
    dst[1, 16]floatND
    核函数名acosh_custom
  • 样例实现:
    本样例中实现的是固定shape为输入src[1, 16],输出dst[1, 16]的acosh_custom。

    • Kernel实现

      使用Acosh高阶API接口完成反双曲余弦计算。

    • Tiling实现

      Host侧通过GetAcoshMaxMinTmpSize获取Acosh接口计算所需的最大和最小临时空间。

    • 调用实现 使用内核调用符<<<>>>调用核函数。

编译运行

在本样例根目录下执行如下步骤,编译并执行样例。

  • 配置环境变量
    请根据当前环境上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-2201 ..;make -j;    # 编译工程,默认npu模式
    ./demo                           # 执行编译生成的可执行程序,执行样例
    

    使用 CPU调试 或 NPU仿真 模式时,添加 -DCMAKE_ASC_RUN_MODE=cpu-DCMAKE_ASC_RUN_MODE=sim 参数即可。

    示例如下:

    cmake -DCMAKE_ASC_RUN_MODE=cpu -DCMAKE_ASC_ARCHITECTURES=dav-2201 ..;make -j; # cpu调试模式
    cmake -DCMAKE_ASC_RUN_MODE=sim -DCMAKE_ASC_ARCHITECTURES=dav-2201 ..;make -j; # NPU仿真模式
    

    注意: 切换编译模式前需清理 cmake 缓存,可在 build 目录下执行 rm CMakeCache.txt 后重新 cmake。

  • 编译选项说明

    选项 可选值 说明
    CMAKE_ASC_RUN_MODE npu(默认)、cpusim 运行模式:NPU 运行、CPU调试、NPU仿真
    CMAKE_ASC_ARCHITECTURES dav-2201(默认)、dav-3510 NPU 架构:dav-2201 对应 Atlas A2/A3 系列,dav-3510 对应 Ascend 950PR/Ascend 950DT
  • 执行结果

    执行结果如下,说明精度对比成功。

    test pass!