auto_mixed_precision_pass.cc modify use_mkldnn [fluid_ops] (#74549)

Author: co63oc

paddle/fluid/framework/ir/auto_mixed_precision_pass.cc

@@ -137,7 +137,7 @@ void DoInsertCastOp(Graph* graph,
       desc.SetAttr("in_dtype", in_dtype);
       desc.SetAttr("out_dtype", out_dtype);
     }
-    desc.SetAttr("use_mkldnn", false);
+    desc.SetAttr("use_onednn", false);
     desc.SetAttr("with_quant_attr", false);
     desc.Flush();
   };

paddle/fluid/framework/ir/pass_test_util.cc

@@ -34,10 +34,10 @@ OpDesc* CreateOp(ProgramDesc* prog,
                  const std::string& op_type_name,
                  const std::vector<InOutVarNamePair>& inputs,
                  const std::vector<InOutVarNamePair>& outputs,
-                 bool use_mkldnn) {
+                 bool use_onednn) {
   auto* op = prog->MutableBlock(0)->AppendOp();
   op->SetType(op_type_name);
-  op->SetAttr("use_mkldnn", use_mkldnn);
+  op->SetAttr("use_onednn", use_onednn);
 
   for (const auto& input : inputs) {
     op->SetInput(input.first, {input.second});

paddle/fluid/framework/ir/pass_test_util.h

@@ -46,16 +46,16 @@ using OpTypeCountPair = std::pair<std::string, int>;
 /// @param[in]  inputs        The vector of input pairs: {input_name, variable
 ///                           name}
 /// @param[in]  outputs       The vector of output pairs {output_name, variable}
-/// @param[in]  use_mkldnn    The flag deciding whether or not to set
-///                           'use_mkldnn' attribute.
+/// @param[in]  use_onednn    The flag deciding whether or not to set
+///                           'use_onednn' attribute.
 ///
 /// @return     Returns pointer to the created operator descriptor.
 ///
 OpDesc* CreateOp(ProgramDesc* prog,
                  const std::string& op_type_name,
                  const std::vector<InOutVarNamePair>& inputs,
                  const std::vector<InOutVarNamePair>& outputs,
-                 bool use_mkldnn = true);
+                 bool use_onednn = true);
 
 ///
 /// @brief      Check whether node 'to' is reachable from node 'from' in graph.

paddle/fluid/framework/ir/pass_tester_helper.h

@@ -286,22 +286,22 @@ struct Layers {
                VarDesc* out = nullptr,
                int x_num_col_dims = 1,
                int y_num_col_dims = 1,
-               bool use_mkldnn = false) {
+               bool use_onednn = false) {
     AttributeMap attrs;
     attrs["x_num_col_dims"] = x_num_col_dims;
     attrs["y_num_col_dims"] = y_num_col_dims;
-    attrs["use_mkldnn"] = use_mkldnn;
+    attrs["use_onednn"] = use_onednn;
     return binary_op("mul", x, y, out, &attrs);
   }
 
   VarDesc* elementwise_add(VarDesc* x,
                            VarDesc* y,
                            VarDesc* out = nullptr,
                            int axis = -1,
-                           bool use_mkldnn = false) {
+                           bool use_onednn = false) {
     AttributeMap attrs;
     attrs["axis"] = axis;
-    attrs["use_mkldnn"] = use_mkldnn;
+    attrs["use_onednn"] = use_onednn;
     return binary_op("elementwise_add", x, y, out, &attrs);
   }