意图借助Mind Studio,基于昇腾处理器,利用TBE的DSL以及TIK的开发方式完成对Segmean算子的开发和性能优化。
对于Segmean算子,我们预计使用Python初步完成对其功能的实现,并在之后利用各种数学方法完成对其的性能优化。具体的开发过程中可以参考TensorFlow相关的原型实现。
昇腾处理器是使用了基于Davinci架构的AI Core计算核心的AI芯片。TBE(Tensor Boost Engine)张量加速引擎则为我们提供了开发自定义算子的方式DSL,偏向于从数学角度完成算子的实现。TIK则是Python的算子开发模块,可以更好地从计算机的角度完成算子的实现。
学习笔记version1 涉及到TBE的基本知识、概念以及三种开发场景