MergeMode#
产品支持情况#
Ascend 950PR/Ascend 950DT:支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持
功能说明#
MergeMode定义了VF计算指令中mask未选中元素(非活跃元素)在目标寄存器中的处理方式。适用于vf.add、vf.sub、vf.mul、vf.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)