SetLoadDataRepeat
功能说明
用于设置Load3Dv2接口的repeat参数。设置repeat参数后,可以通过调用一次Load3Dv2接口完成多个迭代的数据搬运。
函数原型
__aicore__ inline void SetLoadDataRepeat(const LoadDataRepeatParam& repeatParams)
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
repeatParams |
输入 |
repeat相关参数,类型为LoadDataRepeatParam,结构体具体定义为: struct LoadDataRepeatParam { uint16_t repeatStride = 0; uint8_t repeatTime = 1; uint8_t repeatMode = 0; }; 参数说明请参考表2。 |
参数名称 |
含义 |
---|---|
repeatTime |
height/width方向上的迭代次数,取值范围:repeatTime ∈[0, 255] 。默认值为1。 |
repeatStride |
height/width方向上的前一个迭代与后一个迭代起始地址的距离,取值范围:n∈[0, 65535],默认值为0。 如果repeatMode为0,repeatStride的单位为16个元素;如果repeatMode为1,repeatStride的单位为32/sizeof(data_type)个元素 ,data_type指Load3Dv2中源操作数的数据类型。 |
repeatMode |
控制repeat迭代的方向,取值范围:k∈[0, 1] 。默认值为0。 0:迭代沿height方向; 1:迭代沿width方向。 |
支持的型号
Atlas A2训练系列产品/Atlas 800I A2推理产品
调用示例
参考调用示例
父主题: 矩阵计算