TbeBatchMatmulElementWiseFusionPass

融合模式

该融合将满足如下Pattern关系的子图中BatchMatMul/GEMM和ElemWise进行UB融合。

模式一：

模式二：

模式三：

使用约束

• 不支持动态shape。
• 模式一Elemwise只支持FusedMulAdd, Add, Div, RealDiv, Relu, ReluGrad，Elemwise1只支持Add, Relu, FusedMulAdd。
• 模式二Elemwise只支持FusedMulAdd, Add, Div, RealDiv, Relu, ReluGrad。
• 模式三Elemwise只支持Mul, Elemwise1只支持Mul, Elemwise2 只支持Sigmoid。
• BatchMatMul支持BatchMatMul, BatchMatMulV2。

支持的型号

Atlas 200/300/500 推理产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品