CANN训练营第三季-昇腾CANN算子精讲课-基于TBE DSL方式开发算子

86 阅读1分钟
CANN训练营第三季 -昇腾CANN算子精讲课-基于TBE DSL方式开发算子****

 

环境准备****


CANN训练营2022年度第三季——镜像申请:

CANN训练营2022年度第三季——镜像申请 - 华为云问卷系统 (huaweicloud.com)

 

image.png 镜像如下:****

****


image.png 代金券申请:

www.hiascend.com/zh/activiti…

 

image.png  

 

 

代金券到账情况:****

 

image.png  

切换到 华北-北京四 申请服务器

 

image.png 配置网络后设置root用户的登录密码

 

image.png  

通过shell登录服务器如下所示:

 

image.png  

image.png

 

 

算子开发****


算子描述****
使用Mind Studio开发工具,基于TBE DSL方式开发一个Sinh算子。
算子规格参考如下:
- 算子类型:Sinh
‘- 数学表达式:sinh(x)=(exp(x) - exp(-x)) / 2.0
- 支持的数据类型:float16
- 支持的数据排布格式:ND

算子实现文件sinh.py

****

image.png


算子原型定义文件:sinh.h

****

image.png


算子信息库:sinh.ini

****

image.png


算子适配插件:sinh.cc

****

image.png