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

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

环境准备

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

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

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

 

镜像如下:

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

 

代金券申请:

昇腾社区-官网丨昇腾万里 让智能无所不及

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

代金券到账情况:

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

 

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

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

 

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

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

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

 

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

 

算子开发

算子描述

使用Mind Studio开发工具,基于TBE DSL方式开发一个Sinh算子。

算子规格参考如下:

- 算子类型:Sinh

‘- 数学表达式:sinh(x)=(exp(x) - exp(-x)) / 2.0

- 支持的数据类型:float16

- 支持的数据排布格式:ND

算子实现文件sinh.py

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

 

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

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

 

算子信息库:sinh.ini

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

 

算子适配插件:sinh.cc

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

 

你可能感兴趣的:(服务器,运维)