REGISTER_TILING_DEFAULT

产 品 支 持 情 况

产 品	是 否 支 持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品	√
Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品	√
Atlas 200I/500 A2 推 理 产 品	√
Atlas 推 理 系 列 产 品AI Core	√
Atlas 推 理 系 列 产 品Vector Core	√
Atlas 训 练 系 列 产 品	x

功 能 说 明

用 于 在kernel侧 注 册 用 户 使 用 标 准C++语 法 自 定 义 的 默 认TilingData结 构 体。

注 册TilingData结 构 体 用 于 告 知 框 架 侧 用 户 使 用 标 准C++语 法 来 定 义TilingData，同 时 告 知 框 架TilingData结 构 体 类 型，用 于 框 架 做tiling数 据 解 析。

函 数 原 型

Text

REGISTER_TILING_DEFAULT(TILING_STRUCT)

参 数 说 明

参 数	输 入/输 出	说 明
TILING_STRUCT	输 入	用 户 注 册 的 默 认 自 定 义TilingData结 构 体。

约 束 说 明

• 若TilingData结 构 体 在 命 名 空 间 内，注 册 时 需 要 携 带 对 应 的 命 名 空 间 作 用 域 符。
• 暂 不 支 持Kernel直 调 工 程。

调 用 示 例

Text

extern "C" __global__ __aicore__ void add_custom(__gm__ uint8_t *x, __gm__ uint8_t *y, __gm__ uint8_t *z, __gm__ uint8_t *tiling)
{
    REGISTER_TILING_DEFAULT(optiling::TilingData); // 用 于 在kernel侧 注 册 用 户 使 用 标 准C++语 法 自 定 义 的 默 认TilingData结 构 体
    GET_TILING_DATA(tilingData, tiling);
    KernelAdd op;
    op.Init(x, y, z, tilingData.blkDim, tilingData.totalSize, tilingData.splitTile);
    op.Process();
}