【2024昇腾&鲲鹏训练营】Ascend C算子开发能力认证(进阶)

Ascend C算子开发能力认证(进阶)

本课程是AscendC算子开发的进阶教程,通过课程讲解及样例实操,帮助你学习如何使用AscendC进行自定义算开发,并在实际应用中调用自定义算子。

题目要求

实现Ascend C算子Sinh,算子命名为SinhCustom,编写其kernel侧代码、host侧代码,并完成aclnn算子调用测试。
相关算法:sinh(x) = (exp(x) - exp(-x)) / 2.0

要求:

  1. 完成Sinh算子kernel侧核函数相关代码补齐。
  2. 完成Sinh算子host侧Tiling结构体成员变量创建,以及Tiling实现函数的补齐。
  3. 要支持Float16类型输入输出。

算子开发

Kernel侧

需要对核函数中的几部分进行修改
文件路径:SinhCustom\op_kernel\sinh_custom.cpp
在这里插入图片描述

分别针对CopyIn、Compute、CopyOut三个部分进行代码补充,完成算子核函数的数据流入、计算、数据回传三部分功能。
在这里插入图片描述

Host侧

需要对先完成结构体的定义
文件路径:SinhCustom\op_host\sinh_custom_tiling.h
在这里插入图片描述
再补充对Tiling块的初始化部分
文件路径:SinhCustom\op_host\sinh_custom.cpp
在这里插入图片描述

测试结果

bash build.sh

在这里插入图片描述

cd build_out
./custom_opp_ubuntu_aarch64.run

在这里插入图片描述

bash run.sh

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值