SetSingleShape

功能说明

设置Matmul单核计算的形状singleCoreM、singleCoreN、singleCoreK，单位为元素个数。用于运行时修改shape，比如复用Matmul对象来处理尾块。与SetTail接口功能一致，建议使用本接口。

函数原型

1

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

参数说明

返回值

无

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 推理系列产品AI Core

Atlas 200I/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.SetSingleShape(tailM,tailN,tailK);    // 如果是尾核，需要调整singleCoreM/singleCoreN/singleCoreK
mm.IterateAll(gm_c);