diff --git a/paddle/fluid/framework/ir/onednn/compute_propagate_scales_onednn_pass_tester.cc b/paddle/fluid/framework/ir/onednn/compute_propagate_scales_onednn_pass_tester.cc index b17c0a1e9bb9c3..acf5190f459e38 100644 --- a/paddle/fluid/framework/ir/onednn/compute_propagate_scales_onednn_pass_tester.cc +++ b/paddle/fluid/framework/ir/onednn/compute_propagate_scales_onednn_pass_tester.cc @@ -206,7 +206,7 @@ void SetOp(ProgramDesc* prog, const std::unordered_map& attrs = {}) { auto* op = prog->MutableBlock(0)->AppendOp(); op->SetType(type); - op->SetAttr("use_mkldnn", true); + op->SetAttr("use_onednn", true); op->SetAttr("name", name); if (!attrs.empty()) for (auto& attr : attrs) op->SetAttr(attr.first, attr.second); diff --git a/paddle/fluid/ir_adaptor/translator/op_translator.cc b/paddle/fluid/ir_adaptor/translator/op_translator.cc index 0467a8d141265f..c562c652cd354b 100644 --- a/paddle/fluid/ir_adaptor/translator/op_translator.cc +++ b/paddle/fluid/ir_adaptor/translator/op_translator.cc @@ -266,7 +266,7 @@ inline std::string GetPrefix(pir::IrContext* ctx, const OpDesc& op_desc) { paddle::dialect::IsOneDNNOnlyOp(op_desc.Type())) { if (!HasOpInfo(ctx, op_desc, kOneDNNTargetDialectPrefix)) { VLOG(3) << op_desc.Type() - << "'s use_mkldnn == True, but PIR not support OneDNN for this " + << "'s use_onednn == True, but PIR not support OneDNN for this " "op right now."; return kTargetDialectPrefix; } else {