GET_TILING_DATA_MEMBER

产 品 支 持 情 况

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

功 能 说 明

用 于 获 取tiling结 构 体 的 成 员 变 量。

函 数 原 型

Text

GET_TILING_DATA_MEMBER(struct_name, mem_name, tiling_data, tiling_arg)

参 数 说 明

参 数	输 入/输 出	说 明
struct_name	输 入	指 定 的 结 构 体 名 称。
mem_name	输 入	指 定 的 成 员 变 量 名 称。
tiling_data	输 出	返 回 指 定Tiling结 构 体 的 成 员 变 量。
tiling_arg	输 入	此 参 数 为 算 子 入 口 函 数 处 传 入 的tiling参 数。

约 束 说 明

• 本 函 数 需 在 算 子kernel代 码 处 使 用，并 且 传 入 的tiling_data参 数 不 需 要 声 明 类 型。
• 暂 不 支 持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)
{
    KernelAdd op;
    if ASCEND_IS_AIV {
        GET_TILING_DATA(tilingData, tiling);   // Vector侧 使 用 算 子 默 认 注 册 的 完 整 结 构 体
	op.Init(x, y, z, tilingData.totalLength, tilingData.tileNum);
        op.Process();
    } else {
        GET_TILING_DATA_MEMBER(Add_Struct, tCubeTiling, tCubeTilingVar, tiling); // Cube侧 仅 使 用 算 子 注 册 结 构 体 的 成 员 变 量tCubeTiling
	op.Init(x, y, z, tCubeTilingVar);
	op.Process();
    }
}