SetLoopModePara

Ascend 950PR/Ascend 950DT

√

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

x

Atlas A2 训练系列产品 / Atlas A2 推理系列产品

x

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品 AI Core

x

Atlas 推理系列产品 Vector Core

x

Atlas 训练系列产品

x

loopParams

输入

循环模式参数 LoopModeParams类型，定义如下，具体参数说明请参考表2。

struct LoopModeParams {
        loop1Size = 0;
        loop2Size = 0;
        loop1SrcStride = 0;
        loop1DstStride = 0;
        loop2SrcStride = 0;
        loop2DstStride = 0 ;
};

type

输入

数据搬运模式。DataCopyMVType为枚举类型，定义如下，具体参数说明请参考表3。

enum class DataCopyMVType : uint8_t {
    UB_TO_OUT = 0,
    OUT_TO_UB = 1,
};

loop1Size

用于设置内层循环的循环次数，数据类型为uint32_t，取值范围为[0, 2^21)。

loop2Size

用于设置外层循环的循环次数，数据类型为uint32_t，取值范围为[0, 2^21)。

loop1SrcStride

用于设置内层循环中相邻迭代源操作数的数据块间的间隔，单位为Byte，数据类型为uint64_t。

• 当数据搬运模式是UB_TO_OUT的时候取值范围为[0, 2^21)，并且loop1SrcStride必须32B对齐。
• 当数据搬运模式是OUT_TO_UB的时候取值范围为[0, 2^40)。

loop1DstStride

用于设置内层循环中相邻迭代目的操作数的数据块间的间隔，单位为Byte，数据类型为uint64_t。

• 当数据搬运模式是UB_TO_OUT的时候取值范围为[0, 2^40)。
• 当数据搬运模式是OUT_TO_UB的时候取值范围为[0, 2^21)，并且loop1DstStride必须32B对齐。

loop2SrcStride

用于设置外层循环中相邻迭代源操作数的数据块间的间隔，单位为Byte，数据类型为uint64_t。

• 当数据搬运模式是UB_TO_OUT的时候取值范围为[0, 2^21)，并且loop2SrcStride必须32B对齐。
• 当数据搬运模式是OUT_TO_UB的时候取值范围为[0, 2^40)。

loop2DstStride

用于设置外层循环中相邻迭代目的操作数的数据块间的间隔，单位为Byte，数据类型为uint64_t。

• 当数据搬运模式是UB_TO_OUT的时候取值范围为[0, 2^40)。
• 当数据搬运模式是OUT_TO_UB的时候取值范围为[0, 2^21)，并且loop2DstStride必须32B对齐。

UB_TO_OUT

从UB搬运到GM的通路。

OUT_TO_UB

从GM搬运到UB的通路。