MaskWidth

产品支持情况

• Ascend 950PR/Ascend 950DT：支持

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

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

功能说明

MaskWidth定义了vf.get_mask_spr读取SPR掩码时的位宽展开模式，用于控制从MASK寄存器读取掩码数据时的位扩展粒度。

原型定义

class MaskWidth(enum.Enum):
     B32 = ...  # 读取64位MASK0，每位展开为4位（movp_b32，默认）
     B16 = ...  # 读取128位{MASK1,MASK0}，每位展开为2位（movp_b16）

调用示例

import pypto_pro.language as pl

@pl.vector_function
def vf_kernel():
    preg = vf.get_mask_spr(width=pl.MaskWidth.B32)