Skip to content

Commit 573e68e

Browse files
authored
Merge pull request #14348 from luotao1/speedup_analysis
skip mkldnn related pass when use_mkldnn=false to speedup analysis
2 parents ff28b1f + 433fc7c commit 573e68e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddle/fluid/inference/analysis/analyzer.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,9 @@ void Analyzer::Run(Argument* argument) {
113113
passes.push_back("infer_clean_graph_pass");
114114
passes.push_back("graph_viz_pass"); // add graphviz for debug.
115115
for (auto& pass : ir_passes_) {
116-
if (!disabled_ir_passes_.count(pass)) {
116+
// 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) {
117119
passes.push_back(pass);
118120
passes.push_back("graph_viz_pass"); // add graphviz for debug.
119121
}

0 commit comments

Comments
 (0)