[PIR][oneDNN] Avoid repeat warning logs (#64433)

LLee233 · web-flow · commit 68309bb647c5 · 2024-05-20T16:14:49.000+08:00
diff --git a/paddle/fluid/inference/api/paddle_pass_builder.cc b/paddle/fluid/inference/api/paddle_pass_builder.cc
@@ -624,32 +624,36 @@ const std::vector<std::string> kPirXpuPasses{// Functional pass
                                              "conv2d_bn_xpu_fuse_pass",
                                              "group_norm_silu_xpu_fuse_pass"};
 
-const std::vector<std::string> kPirMkldnnPasses{
-    "depthwise_conv_onednn_pass",
-    "squeeze_transpose_onednn_fuse_pass",
-    "conv2d_bias_fuse_pass",
-    "conv2d_transpose_bias_fuse_pass",
-    "conv3d_bias_fuse_pass",
-    "batch_norm_act_fuse_pass",
-    "scale_matmul_fuse_pass",
-    "reshape_transpose_matmul_fuse_pass",
-    "matmul_transpose_reshape_fuse_pass",
-    "matmul_elementwise_add_fuse_pass",
-    "matmul_activation_fuse_pass",
-    "matmul_add_act_fuse_pass",
-    "fc_onednn_enable_pass",
-    "fc_activation_fuse_pass",
-    "self_attention_fuse_pass",
-    "softplus_activation_fuse_pass",
-    "shuffle_channel_detect_pass",
-    "operator_reshape_onednn_fuse_pass",
-    "conv_elementwise_add_onednn_fuse_pass",
-    "conv_activation_onednn_fuse_pass",
-    "conv_concat_activation_onednn_fuse_pass",
-    "elementwise_act_onednn_fuse_pass",
-    "operator_unsqueeze_onednn_fuse_pass",
-    "operator_scale_onednn_fuse_pass",
-    "onednn_placement_pass"};
+const std::vector<std::string> kPirMkldnnPasses {
+  "depthwise_conv_onednn_pass",              //
+      "squeeze_transpose_onednn_fuse_pass",  //
+      "conv2d_bias_fuse_pass",               //
+      "conv2d_transpose_bias_fuse_pass",     //
+      "conv3d_bias_fuse_pass",               //
+      "batch_norm_act_fuse_pass",            //
+      "scale_matmul_fuse_pass",              //
+      "reshape_transpose_matmul_fuse_pass",  //
+      "matmul_transpose_reshape_fuse_pass",  //
+      "matmul_elementwise_add_fuse_pass",    //
+      "matmul_activation_fuse_pass",         //
+      "matmul_add_act_fuse_pass",            //
+      "fc_onednn_enable_pass",               //
+      "fc_activation_fuse_pass",             //
+#if defined(PADDLE_WITH_AVX512F) && defined(PADDLE_WITH_MKLML) && \
+    defined(PADDLE_WITH_DNNL)
+      "self_attention_fuse_pass",  //
+#endif
+      "softplus_activation_fuse_pass",            //
+      "shuffle_channel_detect_pass",              //
+      "operator_reshape_onednn_fuse_pass",        //
+      "conv_elementwise_add_onednn_fuse_pass",    //
+      "conv_activation_onednn_fuse_pass",         //
+      "conv_concat_activation_onednn_fuse_pass",  //
+      "elementwise_act_onednn_fuse_pass",         //
+      "operator_unsqueeze_onednn_fuse_pass",      //
+      "operator_scale_onednn_fuse_pass",          //
+      "onednn_placement_pass"                     //
+};
 
 const std::vector<std::string> kPirCpuPasses{};
 
diff --git a/paddle/fluid/pir/transforms/passes.h b/paddle/fluid/pir/transforms/passes.h
@@ -56,7 +56,9 @@ USE_PIR_PASS(matmul_elementwise_add_fuse_pass);
 USE_PIR_PASS(matmul_activation_fuse_pass);
 USE_PIR_PASS(fc_onednn_enable_pass);
 USE_PIR_PASS(fc_activation_fuse_pass);
+#if defined(PADDLE_WITH_AVX512F) && defined(PADDLE_WITH_MKLML)
 USE_PIR_PASS(self_attention_fuse_pass);
+#endif
 USE_PIR_PASS(softplus_activation_fuse_pass);
 USE_PIR_PASS(shuffle_channel_detect_pass);
 USE_PIR_PASS(operator_reshape_onednn_fuse_pass);
