设置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推理产品
无