Iterate

功能说明

每调用一次Iterate,会计算出一片baseM * baseN的C矩阵。接口内部会维护迭代进度,调用一次后会对A、B矩阵首地址进行偏移。默认以先M轴再N轴的迭代顺序,也可以通过调整tiling参数iterateOrder,转换为先N轴再M轴的迭代顺序。

当传入数据未对齐,存在尾块时,会在最后一次迭代输出尾块的计算结果。

函数原型

__aicore__ inline bool Iterate(bool enPartialSum = false);

参数说明

参数名

输入/输出

描述

enPartialSum

输入

是否将矩阵乘的结果累加于现有的CO1数据,默认值false。在L0C累加时,只支持A矩阵和B矩阵相乘的输出C矩阵规格为singleM==baseM &&singleN==baseN。(单matmul实例场景,多matmul场景不支持该特性)

返回值

false:单核上的所有数据全部算完。

true:数据仍在迭代计算中。

支持的处理器型号

Atlas A2训练系列产品

注意事项