GetCmpMask

功能说明

此接口用于获取Compare（结果存入寄存器）指令的比较结果。

Compare（结果存入寄存器）指令会将比较后的结果写入CmpMask寄存器中，使用GetCmpMask接口可以获取到CmpMask寄存器的值从而得到Compare的结果。

函数原型

template<typename T> __aicore__ inline void GetCmpMask(const LocalTensor<T>& dst)

参数说明

返回值

无

支持的型号

Atlas推理系列产品（Ascend 310P处理器）AI Core

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

约束说明

无

调用示例

Compare（结果存入寄存器）指令的结果使用uint8_t类型数据存储，因此dstLocal使用uint8_t类型。

LocalTensor<float> src0Local;
LocalTensor<float> src1Local;
LocalTensor<uint8_t> dstLocal;
uint64_t mask = 256 / sizeof(float); // 256为每个迭代处理的字节数，结果为64
BinaryRepeatParams repeatParams = { 1, 1, 1, 8, 8, 8 };
Compare(src0Local, src1Local, CMPMODE::LT, mask, repeatParams);
GetCmpMask(dstLocal);