昇腾AI软件栈提供了TBE算子开发框架,开发者可以基于此框架使用Python语言开发自定义算子。那么,我们来了解一下什么是TBE。
TBE的全称为Tensor Boost Engine,即张量加速引擎,是一款华为自研的算子开发工具,用于开发能够运行在NPU(Neural-network Processing Unit:神经网络处理器)上的TBE算子,该工具是在业界著名的 开源 项目TVM(Tensor Virtual Machine)基础上扩展的,提供了一套Python API来实施开发活动。
TBE的优势特性大致如下: