TilingParse
函数功能
注册算子的TilingParse函数,用于解析算子编译阶段生成的算子信息json文件,在注册时需要注册算子自行指定数据类型T,该数据类型用于保存解析后的算子信息。
用户需要为算子编写一个KernelFunc类型或者TilingParseFunc类型的函数,并使用下列对应的接口进行注册。
KernelFunc类型定义如下:
using KernelFunc = UINT32 (*)(KernelContext *context);
TilingParseFunc类型定义如下:
using TilingParseFunc = UINT32 (*)(TilingParseContext *context);
函数原型
template<typename T>
OpImplRegisterV2 &TilingParse(KernelFunc const tiling_parse_func);
template<typename T>
OpImplRegisterV2 &TilingParse(TilingParseFunc const tiling_parse_func);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
tiling_parse_func |
输入 |
待注册的TilingParse函数,类型支持2种:KernelFunc、TilingParseFunc。 |
返回值说明
返回算子的OpImplRegisterV2对象,该对象新增注册了TilingParse函数。
约束说明
无
父主题: OpImplRegisterV2