diff --git a/paddle/fluid/operators/listen_and_serv_op.cc b/paddle/fluid/operators/listen_and_serv_op.cc index 088366dac7b2842dc06daa078e739fa42061a7d7..f235c86ad56999ed84eafba163487a854666d32d 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,