Skip to content

Commit

Permalink
correct condition
Browse files Browse the repository at this point in the history
  • Loading branch information
b3602sss committed Jun 3, 2021
1 parent 2d2999e commit 5bffd93
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion paddle/fluid/inference/api/analysis_predictor.cc
Original file line number Diff line number Diff line change
Expand Up @@ -298,7 +298,8 @@ static void DisablePrepareDataOpt(
}
if (op->HasAttr("sub_block")) {
int blockID = op->GetBlockAttrId("sub_block");
DisablePrepareDataOpt(inference_program, blockID, disable_opt);
DisablePrepareDataOpt(inference_program, blockID,
disable_opt || pre_disable_opt);
}
if (IsPrepareDataOptTargetOp(op)) {
disable_opt = true;
Expand Down

0 comments on commit 5bffd93

Please sign in to comment.