TbeReduceElemwiseFusionPass
融合模式
该融合将ElemWise + CommReduce类算子做UB融合。
模式一:
模式二:
使用约束
模式一的约束如下:
- 只支持静态场景。
- Elewise1必须有且数量为1;必须保证有且仅有一个CommReduce节点。
- Elewise2、3节点数量可无,最多分别不超5。
- 融合时当elsewise不支持从多输入节点作为头开始匹配,也不支持该多输入的elewise节点作为中间节点。
模式二的约束如下::
- 只支持静态场景
- 必须保证有且仅有一个CommReduce节点。
- Elewis4节点数量可无,最多不超多5。
- Elsewise2不能为多输入节点。
父主题: UB融合规则说明