MergeMode#

产品支持情况#

  • Ascend 950PR/Ascend 950DT:支持

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

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

功能说明#

MergeMode定义了VF计算指令中mask未选中元素(非活跃元素)在目标寄存器中的处理方式。适用于vf.addvf.subvf.mulvf.div等VF计算接口。

原型定义#

class MergeMode(enum.Enum):
     ZEROING = ...  # mask未选中位置置零(默认)
     MERGING = ...  # mask未选中位置保留目标寄存器原值

调用示例#

import pypto_pro.language as pl

@pl.vector_function
def vf_kernel():
    dst = vf.add(src0, src1, preg, mode=pl.MergeMode.ZEROING)