GeGluV2FusionPass
融合模式
完成如下结构的融合,把小算子(SplitVD + Gelu + Mul)融合成大算子GeGluV2。
使用约束
仅支持SplitVD节点切分轴大于512的场景。
支持的型号
Atlas 推理系列产品
Atlas A2训练系列产品/Atlas 800I A2推理产品中的AscendxxxB。
父主题: 图融合规则说明
完成如下结构的融合,把小算子(SplitVD + Gelu + Mul)融合成大算子GeGluV2。
仅支持SplitVD节点切分轴大于512的场景。
Atlas 推理系列产品
Atlas A2训练系列产品/Atlas 800I A2推理产品中的AscendxxxB。