跳到主要内容

Tiling

函数功能

注册算子的Tiling函数。

开发者需要为算子编写一个TilingKernelFunc类型的函数,并使用该接口进行注册。同时可以指定tiling数据的最大长度,缺省值为2048Bytes。

TilingKernelFunc类型定义如下。

using TilingKernelFunc = UINT32 (*)(TilingContext *);

函数原型

OpImplRegisterV2 &Tiling(TilingKernelFunc tiling_func, size_t max_tiling_data_size = 2048);

参数说明

参数输入/输出说明
tiling_func输入待注册的自定义Tiling函数,类型为TilingKernelFunc。
max_tiling_data_size输入tiling数据的最大长度,默认值为2048Bytes。

返回值

返回算子的OpImplRegisterV2对象,该对象新增注册了Tiling函数tiling_func。

约束说明