下载
中文
注册

TbeReduceElemwiseFusionPass

融合模式

该融合将ElemWise + CommReduce类算子做UB融合。

模式一:

模式二:

使用约束

模式一的约束如下:

  • 只支持静态场景。
  • Elewise1必须有且数量为1;必须保证有且仅有一个CommReduce节点。
  • Elewise2、3节点数量可无,最多分别不超5。
  • 融合时当elsewise不支持从多输入节点作为头开始匹配,也不支持该多输入的elewise节点作为中间节点。

模式二的约束如下:

  • 只支持静态场景
  • 必须保证有且仅有一个CommReduce节点。
  • Elewis4节点数量可无,最多不超多5。
  • Elsewise2不能为多输入节点。