pypto_pro.language.ReluPreMode#

产品支持情况#

  • Ascend 950PR/Ascend 950DT:支持

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

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

功能说明#

使能随路ReLU操作的枚举,用于L0C Buffer->UB数据搬运场景。

随路ReLU逐元素将负值置零、正值保持不变,计算公式如下:

\[\begin{split}dst = \max(src, 0) = \begin{cases} src & src > 0 \\ 0 & src \leq 0 \end{cases}\end{split}\]

原型定义#

PYPTO_DECLARE_ENUM(ReluPreMode,
    NormalRelu      # 使能随路ReLU操作
)