fix ci coverage

PaddlePaddle · Jun 4, 2021 · 16d776d · 16d776d · paddle-bot-old · Jun 4, 2021
1 parent e08d91b
commit 16d776d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/paddle/fluid/inference/api/analysis_predictor.cc b/paddle/fluid/inference/api/analysis_predictor.cc
@@ -292,7 +292,7 @@ static void DisablePrepareDataOpt(
     std::shared_ptr<framework::ProgramDesc> inference_program, int block,
     bool pre_disable_opt) {
   bool disable_opt = false;
-  auto &infer_block = inference_program_->Block(block);
+  auto &infer_block = inference_program->Block(block);
   for (auto *op : infer_block.AllOps()) {
     if (disable_opt || pre_disable_opt) {
       op->SetAttr("inference_force_prepare_data", true);
@@ -302,9 +302,8 @@ static void DisablePrepareDataOpt(
       DisablePrepareDataOpt(inference_program, blockID,
                             disable_opt || pre_disable_opt);
     }
-    if (IsPrepareDataOptTargetOp(op)) {
-      disable_opt = true;
-    }
+    // disable prepare data if unfriendly op is found
+    disable_opt = IsPrepareDataOptTargetOp(op);
   }
 }