算子实现_算子开发和性能优化 学习笔记

57c193874bb36b56329b13d3b673f665.png

意图借助Mind Studio,基于昇腾处理器,利用TBE的DSL以及TIK的开发方式完成对Segmean算子的开发和性能优化。

对于Segmean算子,我们预计使用Python初步完成对其功能的实现,并在之后利用各种数学方法完成对其的性能优化。具体的开发过程中可以参考TensorFlow相关的原型实现。

昇腾处理器是使用了基于Davinci架构的AI Core计算核心的AI芯片。TBE(Tensor Boost Engine)张量加速引擎则为我们提供了开发自定义算子的方式DSL,偏向于从数学角度完成算子的实现。TIK则是Python的算子开发模块,可以更好地从计算机的角度完成算子的实现。

学习笔记version1 涉及到TBE的基本知识、概念以及三种开发场景

af4c0b64550c3f3abe747085dbe0d0d6.png

1571900467112ea7b7bc21e2df8f70bb.png

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值