bug fix

7 years ago · f224948f31
parent 98c30c7cbe
commit f224948f31
1 changed files with 3 additions and 1 deletions
--- a/paddle/fluid/operators/listen_and_serv_op.cc
+++ b/paddle/fluid/operators/listen_and_serv_op.cc
@ -300,8 +300,10 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
  }

  int checkpoint_point_block_id = Attr<int>(kCheckpointBlockId);
+  auto *ctx = new ExecutorPrepareContext(*program, checkpoint_point_block_id);
+
  std::shared_ptr<framework::ExecutorPrepareContext> ckpt_pre_context =
-      executor.Prepare(*program, checkpoint_point_block_id);
+      std::shared_ptr<ExecutorPrepareContext>(ctx);

  auto f =
      std::bind(FillRequestCtx, std::placeholders::_1, &recv_scope, &dev_ctx,