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.
2 parents ff28b1f + 433fc7c commit 573e68eCopy full SHA for 573e68e
paddle/fluid/inference/analysis/analyzer.cc
@@ -113,7 +113,9 @@ void Analyzer::Run(Argument* argument) {
113
passes.push_back("infer_clean_graph_pass");
114
passes.push_back("graph_viz_pass"); // add graphviz for debug.
115
for (auto& pass : ir_passes_) {
116
- if (!disabled_ir_passes_.count(pass)) {
+ // skip mkldnn pass when use_mkldnn_ = false;
117
+ bool skip_pass = (!use_mkldnn_) && pass.find("mkldnn") != std::string::npos;
118
+ if (!disabled_ir_passes_.count(pass) && !skip_pass) {
119
passes.push_back(pass);
120
121
}
0 commit comments