SetSingleShape
功能说明
设置Matmul单核计算的形状singleCoreM、singleCoreN、singleCoreK，单位为元素个数。用于运行时修改shape，比如复用matmul对象来处理尾块。
函数原型
__aicore__ inline void SetSingleShape(int singleCoreM, int singleCoreN, int singleCoreK)
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
singleCoreM |
输入 |
设置的singleCoreM大小，单位为元素。 |
singleCoreN |
输入 |
设置的singleCoreN大小，单位为元素。 |
singleCoreK |
输入 |
设置的singleCoreK大小，单位为元素。 |
返回值
无
支持的型号
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas推理系列产品AI Core
Atlas 200I/500 A2推理产品
注意事项
无
父主题： Matmul