From 1054b23e1d6c11f02e18bc43a4f055ec3bc8f79e Mon Sep 17 00:00:00 2001 From: csy0225 <78470701+csy0225@users.noreply.github.com> Date: Wed, 1 Mar 2023 15:02:52 +0800 Subject: [PATCH] [XPU] Fix xpu_fuse_pass error caused by weight sharing by other operators. (#51039) --- paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc b/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc index 0def7069e67..c758d45622f 100644 --- a/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc +++ b/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc @@ -76,6 +76,7 @@ FcXPUPattern::FcXPUPattern(PDPattern* pattern, auto* mul_w = pattern->NewNode(mul_w_repr()) ->assert_is_op_input(mul_type_, "Y") ->assert_is_persistable_var() + ->assert_has_n_outputs(1) ->assert_more([](Node* node) { return node->Var()->GetShape().size() == 2; }); -- GitLab