昇腾社区首页
中文
注册

GeGluV2FusionPass

融合模式

完成如下结构的融合,把小算子(SplitVD + Gelu + Mul)融合成大算子GeGluV2。

使用约束

仅支持SplitVD节点切分轴大于512的场景。

支持的型号

Atlas 推理系列产品

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

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