TBE算子开发方式

昇腾AI软件栈提供了TBE（Tensor Boost Engine：张量加速引擎）算子开发框架，开发者可以基于此框架使用Python语言开发自定义算子，通过TBE进行算子开发有以下几种方式：

• DSL（Domain-Specific Language）开发

  为了方便开发者进行自定义算子开发，TBE DSL接口已高度封装，开发者仅需要使用DSL接口完成计算过程的表达，后续的Schedule创建、优化及编译都可通过已有接口一键式完成，适合初级开发者。DSL开发的算子性能可能较低。

• TIK（Tensor Iterator Kernel）开发

  TIK（Tensor Iterator Kernel）是一种基于Python语言的动态编程框架，呈现为一个Python模块。开发者可以通过调用TIK提供的API基于Python语言编写自定义算子，即TIK DSL，然后TIK编译器会将TIK DSL编译为昇腾AI处理器应用程序的二进制文件。

  基于TIK的自定义算子开发，提供了对Buffer的管理和数据自动同步机制，但需要开发者手动计算数据的分片和索引，需要开发者对AI Core架构有一定的了解，入门难度更高。TIK对矩阵的操作更加灵活，性能会更优。

整个计算过程可以表现为多个输入张量经过一个计算节点得到多个张量的过程。TIK方式、DSL的开发流程本质上是一样的，只不过开发的抽象层次不一样而已。