昇腾社区首页
中文
注册

AddRmsNormFusionGraphPass

融合模式

该融合将符合图融合pattern的Add算子和RmsNorm算子融合成融合算子AddRmsNorm,其中Add算子的输出作为RmsNorm算子的第一个输入。

使用约束

  • Add算子的输入X1和X2的shape/dtype要一致。
  • format只支持ND。

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas A3 训练系列产品/Atlas A3 推理系列产品