下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

SetLoadDataRepeat

功能说明

用于设置Load3Dv2接口的repeat参数。设置repeat参数后,可以通过调用一次Load3Dv2接口完成多个迭代的数据搬运。

函数原型

__aicore__ inline void SetLoadDataRepeat(const LoadDataRepeatParam& repeatParams)

参数说明

表1 参数说明

参数名称

输入/输出

含义

repeatParams

输入

repeat相关参数,类型为LoadDataRepeatParam,结构体具体定义为:

struct LoadDataRepeatParam {
    uint16_t repeatStride = 0;
    uint8_t repeatTime = 1;
    uint8_t repeatMode = 0;
};

参数说明请参考表2

表2 LoadDataRepeatParam结构体参数说明

参数名称

含义

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

调用示例

参考调用示例

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词