0a2e47c1创建于 2025年12月27日历史提交

SetLoadDataRepeat

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

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

__aicore__ inline void SetLoadDataRepeat(const LoadDataRepeatParam& repeatParams)

表 1 参数说明

参数名称	输入/输出	含义
repeatParams	输入	设置Load3Dv2接口的repeat参数，类型为LoadDataRepeatParam。具体定义请参考 ${INSTALL_DIR}/include/ascendc/basic_api/interface/kernel_struct_mm.h，${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。参数说明请参考表2。

参数名称

输入/输出

含义

repeatParams

输入

设置Load3Dv2接口的repeat参数，类型为LoadDataRepeatParam。

具体定义请参考 ${INSTALL_DIR}/include/ascendc/basic_api/interface/kernel_struct_mm.h，${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。

参数说明请参考表2。

表 2 LoadDataRepeatParam结构体参数说明

参数名称	含义
repeatTime	height/width方向上的迭代次数，取值范围：repeatTime ∈[0, 255] 。默认值为1。
repeatStride	height/width方向上的前一个迭代与后一个迭代起始地址的距离，取值范围：n∈[0, 65535]，默认值为0。 repeatMode为0，repeatStride的单位为16个元素。 repeatMode为1，repeatStride的单位和具体型号有关。下文中的data_type指Load3Dv2中源操作数的数据类型。 Atlas A2 训练系列产品/Atlas A2 推理系列产品，repeatStride的单位为32/sizeof(data_type)个元素。 Atlas A3 训练系列产品/Atlas A3 推理系列产品，repeatStride的单位为32/sizeof(data_type)个元素。
repeatMode	控制repeat迭代的方向，取值范围：k∈[0, 1] 。默认值为0。 0：迭代沿height方向； 1：迭代沿width方向。

参数名称

含义

repeatTime

height/width方向上的迭代次数，取值范围：repeatTime ∈[0, 255] 。默认值为1。

repeatStride

height/width方向上的前一个迭代与后一个迭代起始地址的距离，取值范围：n∈[0, 65535]，默认值为0。

repeatMode为0，repeatStride的单位为16个元素。
repeatMode为1，repeatStride的单位和具体型号有关。下文中的data_type指Load3Dv2中源操作数的数据类型。
Atlas A2 训练系列产品/Atlas A2 推理系列产品，repeatStride的单位为32/sizeof(data_type)个元素。

Atlas A3 训练系列产品/Atlas A3 推理系列产品，repeatStride的单位为32/sizeof(data_type)个元素。

repeatMode

控制repeat迭代的方向，取值范围：k∈[0, 1] 。默认值为0。

0：迭代沿height方向；

1：迭代沿width方向。