Skip to content

Commit 7badcdb

Browse files
co63ocmaxiaolong001
authored andcommitted
is_test_pass_tester.cc modify use_mkldnn [fluid_ops] (PaddlePaddle#74518)
1 parent 0b95d33 commit 7badcdb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/fluid/framework/ir/is_test_pass_tester.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,14 +29,14 @@ void SetOp(ProgramDesc* prog,
2929
const std::string& name,
3030
const std::vector<std::string>& inputs,
3131
const std::vector<std::string>& outputs,
32-
bool use_mkldnn = false,
32+
bool use_onednn = false,
3333
ISTEST_STATE is_test = ISTEST_STATE::UNSET) {
3434
auto* op = prog->MutableBlock(0)->AppendOp();
3535
op->SetType(type);
3636
op->SetAttr("name", name);
3737
op->SetInput("X", inputs);
3838
op->SetOutput("Out", outputs);
39-
op->SetAttr("use_mkldnn", use_mkldnn);
39+
op->SetAttr("use_onednn", use_onednn);
4040
if (is_test == ISTEST_STATE::UNSET)
4141
op->MutableAttrMap()->erase("is_test");
4242
else if (is_test == ISTEST_STATE::FALSE)

0 commit comments

Comments
 (0)