Merge pull request #13992 from wojtuss/wojtuss/add-ifdef-mkldnn

Add ifdef guard for MKL-DNN placement pass
7 years ago · 252401c5d3
parent a1d3db031b 4a368a4901
commit 252401c5d3
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/inference/analysis/analyzer.cc
+++ b/paddle/fluid/inference/analysis/analyzer.cc
@ -101,10 +101,12 @@ Analyzer::Analyzer() { Register("manager1", new DfgPassManagerImpl); }

 void Analyzer::Run(Argument* argument) {
  std::vector<std::string> passes;
+#ifdef PADDLE_WITH_MKLDNN
  if (use_mkldnn_) {
    VLOG(3) << "Adding MKL-DNN placement pass";
    passes.push_back("mkldnn_placement_pass");
  }
+#endif
  for (auto& pass : ir_passes_) {
    if (!disabled_ir_passes_.count(pass)) {
      passes.push_back(pass);