InplaceAddRmsNormFusionPass
融合模式
将AddRmsNorm的y输出地址复用x1输入地址,x输出地址复用x2输入地址,转换为InplaceAddRmsNorm算子。
使用约束
无
支持的型号
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 推理系列产品
父主题: 图融合规则说明
将AddRmsNorm的y输出地址复用x1输入地址,x输出地址复用x2输入地址,转换为InplaceAddRmsNorm算子。
无
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 推理系列产品