SetTail
功能说明
在不改变tiling的情况下,重新设置本次计算的singleCoreM/singleCoreN/singleCoreK。
函数原型
__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 200I/500 A2推理产品
注意事项
无
父主题: Matmul