SetLoadDataRepeat产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品 / Atlas A3 推理系列产品√Atlas A2 训练系列产品 / Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品√Atlas 推理系列产品 AI CorexAtlas 推理系列产品 Vector CorexAtlas 训练系列产品x功能说明用于设置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。表 2LoadDataRepeatParam结构体参数说明参数名称含义repeatStrideheight/width方向上的前一个迭代与后一个迭代起始地址的距离取值范围n∈[0, 65535]默认值为0。repeatMode为0repeatStride的单位为16个元素。repeatMode为1repeatStride的单位和具体型号有关。下文中的data_type指Load3Dv2中源操作数的数据类型。Ascend 950PR/Ascend 950DTrepeatStride的单位为32/sizeof(data_type)个元素 。Atlas A2 训练系列产品 / Atlas A2 推理系列产品repeatStride的单位为32/sizeof(data_type)个元素 。Atlas A3 训练系列产品 / Atlas A3 推理系列产品repeatStride的单位为32/sizeof(data_type)个元素 。Atlas 200I/500 A2 推理产品repeatStride的单位为64/sizeof(data_type)个元素。repeatTimeheight/width方向上的迭代次数取值范围repeatTime ∈[0, 255] 。默认值为1。repeatMode控制repeat迭代的方向取值范围k∈[0, 1] 。默认值为0。0迭代沿height方向1迭代沿width方向。调用示例参考调用示例创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考