[XPU] Fix matmul + elementwise_add + bn + act fusion that elementwise…

…_add has branch. (#56578)
PaddlePaddle · Aug 24, 2023 · e9bb312 · e9bb312
1 parent 1c0db09
commit e9bb312
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc b/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc
@@ -128,6 +128,9 @@ FcXPUPattern::FcXPUPattern(PDPattern* pattern,
     add_out = pattern->NewNode(add_out_repr())
                   ->assert_is_op_output("elementwise_add", "Out")
                   ->assert_var_not_persistable();
+    if (with_bn_ || !act_type_.empty()) {
+      add_out->assert_has_n_outputs(1);
+    }
     add->LinksFrom({mul_out, bias}).LinksTo({add_out});
   } else {
     add_out = mul_out;