asc.language.basic.set_load_data_repeat

asc.language.basic.set_load_data_repeat(param: LoadDataRepeatParam) → None

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

对应的 Ascend C 函数原型

__aicore__ inline void SetLoadDataRepeat(const LoadDataRepeatParam& repeatParams)

参数说明

repeatParams
设置load_3d_v22接口的repeat参数，类型为LoadDataRepeatParam。
repeatParams
height/width方向上的迭代次数，取值范围：repeatTime ∈[0, 255] 。默认值为1
repeatStride
height/width方向上的前一个迭代与后一个迭代起始地址的距离，取值范围：n∈[0, 65535]，默认值为0。 repeatMode为0，repeatStride的单位为16个元素。 repeatMode为1，repeatStride的单位和具体型号有关。
repeatMode
控制repeat迭代的方向，取值范围：k∈[0, 1] 。默认值为0。 0：迭代沿height方向； 1：迭代沿width方向。

调用示例

import asc
static_param = asc.LoadDataRepeatParam(
    repeatTime=4,
    repeatStride=8,
    repeatMode=0
)
asc.set_load_data_repeat(static_param)