Registers the Tiling function of an operator.
You need to write a function of the TilingKernelFunc type for the operator and use this API to register the function. You can also specify the maximum length of the tiling data. The default value is 2048 bytes.
The TilingKernelFunc type is defined as follows:
using TilingKernelFunc = UINT32 (*)(TilingContext *);
OpImplRegisterV2 &Tiling(TilingKernelFunc tiling_func, size_t max_tiling_data_size = 2048);
| Parameter | Input/Output | Description |
|---|---|---|
| tiling_func | Input | User-defined Tiling function to be registered, which is of the TilingKernelFunc type. |
| max_tiling_data_size | Input | Maximum length of the tiling data. The default value is 2048 bytes. |