目录

GET_TILING_DATA

Function Description

Retrieves tiling information passed to a kernel entry point function and fills a registered Tiling structure with this data. This function is compiled in macro expansion mode. Note that the TilingData structure needs to be defined in the corresponding operator host implementation to implement and register the Tiling function for TilingData compute. For details, see Tiling Implementation on the Host. If you have registered multiple TilingData structures through TilingData Structure Registration, this API returns the default registered structure.

Function Prototype

GET_TILING_DATA(tiling_data, tiling_arg)

Parameters

Parameter Input/Output Description
tiling_data Output Tiling data, for which the default tiling structure variable is returned.
tiling_arg Input Tiling argument of the entrypoint function of the operator.

Supported Models

Kirin 9020 series processors

Kirin 9030 series processors

Kirin X90 series processors

Restrictions

Calling Example

extern "C" __global__ __aicore__ void add_custom(__gm__ uint8_t *x, __gm__ uint8_t *y, __gm__ uint8_t *z, __gm__ uint8_t *tiling)
{
    GET_TILING_DATA(tilingData, tiling);
    KernelAdd op;
    op.Init(x, y, z, tilingData.blkDim, tilingData.totalSize, tilingData.splitTile);
    op.Process();
}