Skip to content

Commit a88d0e4

Browse files
committed
Fix
1 parent 225deda commit a88d0e4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/fluid/framework/ir/onednn/cpu_bfloat16_placement_pass_tester.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,11 @@ void SetOp(ProgramDesc* prog,
2525
const std::vector<std::string>& inputs,
2626
const std::vector<std::string>& outputs,
2727
const std::string& mkldnn_data_type = "float32",
28-
const bool use_mkldnn = true) {
28+
const bool use_onednn = true) {
2929
auto* op = prog->MutableBlock(0)->AppendOp();
3030

3131
op->SetType(type);
32-
if (type != "reshape2") op->SetAttr("use_mkldnn", use_mkldnn);
32+
if (type != "reshape2") op->SetAttr("use_onednn", use_onednn);
3333
op->SetAttr("mkldnn_data_type", mkldnn_data_type);
3434

3535
if (type == "conv2d") {

0 commit comments

Comments
 (0)