diff --git a/paddle/fluid/framework/ir/is_test_pass_tester.cc b/paddle/fluid/framework/ir/is_test_pass_tester.cc
index 78e9db9e9b8d6..8d594c10392fb 100644
--- a/paddle/fluid/framework/ir/is_test_pass_tester.cc
+++ b/paddle/fluid/framework/ir/is_test_pass_tester.cc
@@ -29,14 +29,14 @@ void SetOp(ProgramDesc* prog,
            const std::string& name,
            const std::vector<std::string>& inputs,
            const std::vector<std::string>& outputs,
-           bool use_mkldnn = false,
+           bool use_onednn = false,
            ISTEST_STATE is_test = ISTEST_STATE::UNSET) {
   auto* op = prog->MutableBlock(0)->AppendOp();
   op->SetType(type);
   op->SetAttr("name", name);
   op->SetInput("X", inputs);
   op->SetOutput("Out", outputs);
-  op->SetAttr("use_mkldnn", use_mkldnn);
+  op->SetAttr("use_onednn", use_onednn);
   if (is_test == ISTEST_STATE::UNSET)
     op->MutableAttrMap()->erase("is_test");
   else if (is_test == ISTEST_STATE::FALSE)