Created by: Xreki
test_ir_fusion_group_pass中,有一个单测对应的子图结构如上。因graph中node的顺序,导致了遍历后节点的顺序不同,因而会产生2种可能的融合结果。
-
结果1(如上图所示):子图1,elementwise_mul+1个红色标记的cast;子图2,elementwise_add+relu+2个绿色标记的cast
-
结果2:子图1,elementwise_mul;子图2,1个红色标记的cast+elementwise_add+relu+2个绿色标记的cast。由于fusion_group_pass中要求子图中的op节点数不小于2,因此两种情况得到的fusion_group op的数量不相同。
这个PR在「红色标记的cast」和elementwise_add直接插入一个softmax,使得子图匹配的结果唯一确定。