UnPack
产品支持情况
- Ascend 950PR/Ascend 950DT：支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品：不支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品：不支持
- Atlas 200I/500 A2 推理产品：不支持
- Atlas 推理系列产品AI Core：不支持
- Atlas 推理系列产品Vector Core：不支持
- Atlas 训练系列产品：不支持
功能说明
头文件路径为："basic_api/reg_compute/kernel_reg_compute_maskreg_intf.h"。
根据所选的低位模式或高位模式，将源操作数src的低半部分或高半部分，展开到目的操作数dst。展开方式为：将每bit展开为2bit，高位置零。
函数原型
C++
template <HighLowPart part = HighLowPart::LOWEST>
__simd_callee__ inline void UnPack(MaskReg& dst, MaskReg& src)
参数说明
表1 模板参数说明
| 参数名 | 描述 |
|---|---|
| part | 枚举类型，低位模式或高位模式。 • LOWEST低位模式； • HIGHEST高位模式。 |
表2 参数说明
| 参数名 | 描述 |
|---|---|
| dst | 目的操作数，类型为MaskReg。 |
| src | 源操作数，类型为MaskReg。 |
返回值说明
无
约束说明
无
调用示例
C++
template <typename T>
__simd_vf__ inline void UnpackVF(__ubuf__ T* dstAddr, __ubuf__ T* srcAddr, uint32_t count, uint32_t oneRepeatSize, uint16_t repeatTimes)
{
AscendC::Reg::RegTensor<T> srcReg;
AscendC::Reg::MaskReg maskFull = AscendC::Reg::CreateMask<T, AscendC::Reg::MaskPattern::ALL>();
AscendC::Reg::MaskReg mask0;
AscendC::Reg::MaskReg mask1;
AscendC::Reg::UnPack<AscendC::Reg::HighLowPart::LOWEST>(mask0, maskFull);
for (uint16_t i = 0; i < repeatTimes; ++i) {
mask1 = AscendC::Reg::UpdateMask<T>(count);
AscendC::Reg::LoadAlign(srcReg, srcAddr + i * oneRepeatSize);
AscendC::Reg::Adds(srcReg, srcReg, 0, mask0);
AscendC::Reg::StoreAlign(dstAddr + i * oneRepeatSize, srcReg, mask1);
}
}