REGISTER_TILING_FOR_TILINGKEY产品支持情况产品是否支持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侧注册与TilingKey相匹配的TilingData自定义结构体该接口需提供一个逻辑表达式逻辑表达式以字符串“TILING_KEY_VAR”代指实际TilingKey表达TilingKey所满足的范围。函数原型REGISTER_TILING_FOR_TILINGKEY(EXPRESSION, TILING_STRUCT)参数说明参数输入/输出说明EXPRESSION输入EXPRESSION为逻辑运算其中用TILING_KEY_VAR指代TilingKey。TILING_STRUCT输入用户注册的与TilingKey相匹配的TilingData自定义结构体。约束说明使用该接口时需确保已使用REGISTER_TILING_DEFAULT注册默认的用户自定义TilingData结构体用于告知框架侧用户使用标准C语法来定义TilingData。EXPRESSION当前支持位运算、|、~、^移位运算符、算术运算、-、*、/、%条件运算符、!、、、、逻辑与、或||以及()。优先级同C。若TilingData结构体在命名空间内注册时需要携带对应的命名空间作用域符。不支持同个TilingKey指向不同TilingData结构体会出现拦截报错。暂不支持kernel直调工程。调用示例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); // 注册用户默认自定义TilingData结构体 REGISTER_TILING_FOR_TILINGKEY(TILING_KEY_VAR 1, optiling::TilingDataA); // 注册TilingKey为1的TilingData结构体 REGISTER_TILING_FOR_TILINGKEY((TILING_KEY_VAR 10) (TILING_KEY_VAR 15), optiling::TilingDataB); // 注册TilingKey在[10,15]之间的TilingData结构体 REGISTER_TILING_FOR_TILINGKEY(TILING_KEY_VAR 0xFF, optiling::TilingDataC); // 注册TilingKey低16位为1的TilingData结构体 if (TILING_KEY_IS(1)) { GET_TILING_DATA_WITH_STRUCT(optiling::TilingDataA, tilingData, tiling); ...... } else if (TILING_KEY_IS(11)) { GET_TILING_DATA_WITH_STRUCT(optiling::TilingDataB, tilingData, tiling); ...... } else if (TILING_KEY_IS(14)) { GET_TILING_DATA_WITH_STRUCT(optiling::TilingDataB, tilingData, tiling); ...... } else if (TILING_KEY_IS(255)) { GET_TILING_DATA_WITH_STRUCT(optiling::TilingDataC, tilingData, tiling); ...... } else { GET_TILING_DATA(tilingData, tiling); ...... } }使用标准C语法注册tiling结构体class TilingDataA{ public: ... }; class TilingDataB{ public: ... }; class TilingDataC{ public: ... };配套的host侧tiling函数示例ge::graphStatus TilingFunc(gert::TilingContext* context) { // 其他代码逻辑 ... if(condition1){ context-SetTilingKey(1); optiling::TilingDataA *Addtiling context-GetTilingDataoptiling::TilingDataA(); ... } else if (condition2){ context-SetTilingKey(11); optiling::TilingDataB *Addtiling context-GetTilingDataoptiling::TilingDataB (); ... } else if (condition3){ context-SetTilingKey(14); optiling::TilingDataB *Addtiling context-GetTilingDataoptiling::TilingDataB (); ... } else if (condition4){ context-SetTilingKey(255); optiling::TilingDataC *Addtiling context-GetTilingDataoptiling::TilingDataC (); ... } ... // 其他代码逻辑 }创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考