From f224948f310963d91d823679cfd6e16d0186ae00 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 15 Jun 2018 16:38:44 +0800 Subject: [PATCH] bug fix --- paddle/fluid/operators/listen_and_serv_op.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/listen_and_serv_op.cc b/paddle/fluid/operators/listen_and_serv_op.cc index 088366dac7..f235c86ad5 100644 --- 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(kCheckpointBlockId); + auto *ctx = new ExecutorPrepareContext(*program, checkpoint_point_block_id); + std::shared_ptr ckpt_pre_context = - executor.Prepare(*program, checkpoint_point_block_id); + std::shared_ptr(ctx); auto f = std::bind(FillRequestCtx, std::placeholders::_1, &recv_scope, &dev_ctx, -- GitLab