We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0b95d33 commit 7badcdbCopy full SHA for 7badcdb
paddle/fluid/framework/ir/is_test_pass_tester.cc
@@ -29,14 +29,14 @@ void SetOp(ProgramDesc* prog,
29
const std::string& name,
30
const std::vector<std::string>& inputs,
31
const std::vector<std::string>& outputs,
32
- bool use_mkldnn = false,
+ bool use_onednn = false,
33
ISTEST_STATE is_test = ISTEST_STATE::UNSET) {
34
auto* op = prog->MutableBlock(0)->AppendOp();
35
op->SetType(type);
36
op->SetAttr("name", name);
37
op->SetInput("X", inputs);
38
op->SetOutput("Out", outputs);
39
- op->SetAttr("use_mkldnn", use_mkldnn);
+ op->SetAttr("use_onednn", use_onednn);
40
if (is_test == ISTEST_STATE::UNSET)
41
op->MutableAttrMap()->erase("is_test");
42
else if (is_test == ISTEST_STATE::FALSE)
0 commit comments