SqueezeMode#

产品支持情况#

  • Ascend 950PR/Ascend 950DT：支持

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

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

功能说明#

SqueezeMode定义了vf.squeeze的数据收集模式，用于控制是否将有效元素的字节计数存储到AR SPR（Special Purpose Register）中。

原型定义#

class SqueezeMode(enum.Enum):
     NO_STORE_REG = ...  # 不将有效元素字节计数存入AR SPR（默认）
     STORE_REG = ...  # 将有效元素总字节数存入AR SPR

调用示例#

import pypto_pro.language as pl

@pl.vector_function
def vf_kernel():
    dst = vf.squeeze(src, preg, mode=pl.SqueezeMode.STORE_REG)