asc_set_mask_norm

产品支持情况

• Ascend 950PR/Ascend 950DT：不支持

• Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持

• Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持

• Atlas 200I/500 A2 推理产品：不支持

• Atlas 推理系列产品AI Core：不支持

• Atlas 推理系列产品Vector Core：不支持

• Atlas 训练系列产品：不支持

功能说明

设置Mask模式为Normal模式，该模式为系统默认模式。该模式下，Mask参数可控制高维切分计算过程中单次迭代内参与计算的元素。

函数原型

C++

__aicore__ inline void asc_set_mask_norm()

参数说明

无

返回值说明

无

流水类型

PIPE_S

约束说明

该API需和asc_set_vector_mask配合使用。

调用示例

C++

asc_set_mask_norm();
asc_set_vector_mask(0xffffffffffffffff, 0xffffffffffffffff);    // 设置每个迭代内所有元素均参与计算。

// 进行高维切分计算
constexpr uint32_t total_length = 128; //total_length指参与计算的数据总长度
__ubuf__ half src0[total_length];
__ubuf__ half src1[total_length];
__ubuf__ half dst[total_length];

uint8_t repeat = 1; // 1个repeat可以处理128个half类型的数据
uint8_t dst_block_stride = 1;
uint8_t src0_block_stride = 1;
uint8_t src1_block_stride = 1;
uint8_t dst_repeat_stride = 8;
uint8_t src0_repeat_stride = 8;
uint8_t src1_repeat_stride = 8;
asc_add(dst, src0, src1, repeat, dst_block_stride, src0_block_stride, src1_block_stride,
    dst_repeat_stride, src0_repeat_stride, src1_repeat_stride);