TbeAippCommonFusionPass
融合模式
该融合将满足如下Pattern关系的子图中Aipp + Conv2D + ElemWise(可选) + AscendQuant(可选) + StridedWrite(可选)对应节点进行UB融合。
ElemWise可以有多个串连,最多支持5个。
使用约束
- Conv2D算子, strides = [1, 1], pad = [0, 0, 0, 0], kernel 1*1 的场景不开启融合。
- Aipp开启resize,不开启融合。
- Aipp mode为dynamic,不开启融合。
- Aipp的input format设置为["RGB16", "RGB20", "RGB24","RGB8_IR", "RGB16_IR","RGB24_IR"]中的一种时,不开启融合。
- Aipp开启padding时,不开启融合。
- 若Conv2D算子使能DMA,不支持融合。
- 若给出最小Tiling,L1仍无法容纳Aipp处理结果,则放弃融合。
- 若卷积的kernal H小于或者等于上下任意一方向的Conv2d pad与AIPP pad之和,则放弃融合。
支持的型号
Atlas 200/300/500 推理产品
Atlas 推理系列产品
Atlas 200/500 A2推理产品
Atlas 训练系列产品
父主题: UB融合规则说明