ZInplaceAddLayerNormFusionPass
融合模式
将AddLayerNorm的y输出地址复用x1输入地址,x输出地址复用x2输入地址,转换为InplaceAddLayerNorm算子。
使用约束
不支持融合前AddLayerNorm算子的输入类型不一致的场景。
支持的型号
Atlas A2训练系列产品/Atlas 800I A2推理产品
父主题: 图融合规则说明
将AddLayerNorm的y输出地址复用x1输入地址,x输出地址复用x2输入地址,转换为InplaceAddLayerNorm算子。
不支持融合前AddLayerNorm算子的输入类型不一致的场景。
Atlas A2训练系列产品/Atlas 800I A2推理产品