概述

TBE自定义算子开发接口包含算子代码实现接口、插件实现接口以及原型定义接口,如下表所示。

表1 算子开发相关接口汇总

接口类型

章节

功能

算子代码实现接口

TBE DSL API

DSL开发方式相关接口。

TBE TIK API

TIK开发方式相关接口。

TBE Platform Info API

昇腾AI处理器硬件信息公共接口。

TBE Utils API

算子开发过程中的通用校验/操作接口。

TBE DSL Testing API

DSL算子的调试接口。

用于在CPU上对DSL算子进行功能调试。

TBE Register API

Register模块提供的算子compute函数注册管理等接口。

TBE Context API

Context模块提供的算子编译期间上下文信息管理功能相关接口。

原型定义接口

原型定义 API

算子原型定义所需接口。

operator API

原型实现时依赖的operator定义相关接口。

插件适配接口

插件适配 API

算子适配插件实现所需接口。

operator API

插件实现时依赖的operator定义相关接口。