SetTail
功能说明
在不改变Tiling的情况下，重新设置本次计算的singleCoreM/singleCoreN/singleCoreK。
函数原型
1 | __aicore__ inline void SetTail(int tailM = -1, int tailN = -1, int tailK = -1) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
tailM |
输入 |
重新设置的singleCoreM值 |
tailN |
输入 |
重新设置的singleCoreN值 |
tailK |
输入 |
重新设置的singleCoreK值 |
返回值
无
支持的型号
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas推理系列产品AI Core
Atlas 200/500 A2推理产品
注意事项
无
调用示例
1 2 3 4 5 6 | REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling); mm.SetTensorA(gm_a); mm.SetTensorB(gm_b); mm.SetBias(gm_bias); mm.SetTail(tailM,tailN,tailK); // 如果是尾核，需要调整singleCoreM/singleCoreN/singleCoreK mm.IterateAll(gm_c); |
父主题： Matmul
