fix KUNLUN-Test CI

PaddlePaddle · May 19, 2023 · 26e1cf3 · 26e1cf3
1 parent 5dc6680
commit 26e1cf3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddle/fluid/framework/ir/xpu/add_activation_xpu_fuse_pass.cc b/paddle/fluid/framework/ir/xpu/add_activation_xpu_fuse_pass.cc
@@ -97,7 +97,8 @@ AddActXPUPattern::AddActXPUPattern(PDPattern* pattern,
                    ->assert_var_not_persistable()
                    ->AsInput();
   auto ele_out = pattern->NewNode(ele_out_repr())
-                     ->assert_is_op_output("elementwise_add", "Out");
+                     ->assert_is_op_output("elementwise_add", "Out")
+                     ->assert_has_n_outputs(1);
   ele_add->LinksFrom({ele_x, ele_y}).LinksTo({ele_out});
   ele_out->assert_is_op_input(act_type_, "X");
   auto act = pattern->NewNode(act_repr())->assert_is_op(act_type_);