SetSingleShape

功能说明

设置Matmul单核计算的形状singleCoreM、singleCoreN、singleCoreK，单位为元素个数。用于运行时修改shape，比如复用Matmul对象来处理尾块。

函数原型

1

__aicore__ inline void SetSingleShape(int singleM, int singleN, int singleK)

参数说明

返回值

无

支持的型号

Atlas A2训练系列产品/Atlas 800I A2推理产品

Atlas推理系列产品AI Core

Atlas 200/500 A2推理产品

注意事项

无

调用示例

REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling);
mm.SetTensorA(gm_a);
mm.SetTensorB(gm_b);
mm.SetBias(gm_bias);
mm.SetSingleShape(tailM,tailN,tailK);    // 如果是尾核，需要调整singleCoreM/singleCoreN/singleCoreK
mm.IterateAll(gm_c);