MaskPattern

产品支持情况

• Ascend 950PR/Ascend 950DT：支持

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

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

功能说明

MaskPattern定义了vf.create_mask和vf.mask_gen_with_reg_tensor的掩码生成模式，用于控制寄存器中哪些元素被标记为有效。

原型定义

class MaskPattern(enum.Enum):
     ALL = ...  # 所有元素有效
     ALLF = ...  # 所有元素无效
     VL1 = ...  # 最低1个元素有效
     VL2 = ...  # 最低2个元素有效
     VL3 = ...  # 最低3个元素有效
     VL4 = ...  # 最低4个元素有效
     VL8 = ...  # 最低8个元素有效
     VL16 = ...  # 最低16个元素有效
     VL32 = ...  # 最低32个元素有效
     VL64 = ...  # 最低64个元素有效
     VL128 = ...  # 最低128个元素有效
     M3 = ...  # 每3个元素中第1个有效
     M4 = ...  # 每4个元素中第1个有效
     H = ...  # 低半部分有效
     Q = ...  # 低四分之一有效

调用示例

import pypto_pro.language as pl

@pl.vector_function
def vf_kernel():
    preg = vf.create_mask(pattern=pl.MaskPattern.VL8)  # 最低8个元素有效