bug fix

paddle/fluid/operators/checkpoint_notify_op.cc

@@ -39,9 +39,10 @@ class CheckpointNotifyOp : public framework::OperatorBase {
 
     detail::RPCClient* rpc_client =
         detail::RPCClient::GetInstance<RPCCLIENT_T>();
-    VLOG(3) << "sending " << ins[i] << " to " << epmap[i] << " to get "
-            << outs[i] << " back";
-    rpc_client->AsyncCheckpointNotify(epmap[i], dir);
+    for (size_t i = 0; i < epmap.size(); i++) {
+      VLOG(3) << "sending to " << epmap[i] << " to checkpoint notify ... ";
+      rpc_client->AsyncCheckpointNotify(epmap[i], dir);
+    }
     rpc_client->Wait();
   }
 };

paddle/fluid/operators/listen_and_serv_op.cc

@@ -300,10 +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);
+  auto ctx = executor.Prepare(*program, checkpoint_point_block_id);
 
   std::shared_ptr<framework::ExecutorPrepareContext> ckpt_pre_context =
-      std::shared_ptr<ExecutorPrepareContext>(ctx);
+      std::move(ctx);
 
   auto f =
       std::bind(FillRequestCtx, std::placeholders::_1, &recv_scope, &dev_ctx,