fix scale_matmul fuse pass (#43089)

e1e0deed · heliqi · GitHub · 1448520d · e1e0deed
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

paddle/fluid/framework/ir/mkldnn/scale_matmul_fuse_pass.cc paddle/fluid/framework/ir/mkldnn/scale_matmul_fuse_pass.cc +4 -0

未找到文件。
--- a/paddle/fluid/framework/ir/mkldnn/scale_matmul_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/mkldnn/scale_matmul_fuse_pass.cc
@@ -91,6 +91,10 @@ void ScaleMatmulFusePass::ApplyImpl(ir::Graph* graph) const {
    GET_IR_NODE_FROM_SUBGRAPH(scale_out, scale_out, scale_matmul_pattern);
    GET_IR_NODE_FROM_SUBGRAPH(matmul_op, matmul_op, scale_matmul_pattern);

+    if ((scale_out->outputs).size() != 1) {
+      return;
+    }
+
    if (scale_op->Op()->GetAttrIfExists<float>("bias") == 0.0) {
      auto matmul_alpha = matmul_op->Op()->GetAttrIfExists<float>("alpha");
      auto scale_scale = scale_op->Op()->GetAttrIfExists<float>("scale");