Fix

Author: co63oc

paddle/fluid/framework/ir/graph_pattern_detector.cc

@@ -3318,12 +3318,7 @@ PDNode *patterns::MKLDNNInPlace::operator()() {
   auto next_output = pattern->NewNode(next_op_out_repr())->AsOutput();
 
   // Check if op is ONE-DNN enabled
-  if (possible_inplace_op->HasAttr("use_mkldnn")) {
-    possible_inplace_op->assert_op_attr("use_mkldnn", true);
-  }
-  if (possible_inplace_op->HasAttr("use_onednn")) {
-    possible_inplace_op->assert_op_attr("use_onednn", true);
-  }
+  possible_inplace_op->assert_op_attr_or("use_mkldnn", "use_onednn", true);
 
   // linked structure
   possible_inplace_op->LinksTo({output});

paddle/fluid/framework/ir/graph_pattern_detector.h

@@ -168,6 +168,19 @@ struct PDNode {
     return this;
   }
 
+  PDNode* assert_op_attr_or(const std::string& attr_name1,
+                            const std::string& attr_name2,
+                            const T& attr) {
+    asserts_.emplace_back([=](Node* x) {
+      return x && x->IsOp() &&
+             ((x->Op()->HasAttr(attr_name1) &&
+               PADDLE_GET_CONST(T, x->Op()->GetAttr(attr_name1)) == attr) ||
+              (x->Op()->HasAttr(attr_name2) &&
+               PADDLE_GET_CONST(T, x->Op()->GetAttr(attr_name2)) == attr));
+    });
+    return this;
+  }
+
  private:
   PDNode(PDPattern* pattern,
          const std::string& name = "",