From 5553adf85ddb1e3839c30f7dd7d89909eabfb8ca Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Mon, 18 Jun 2018 22:45:04 +0800 Subject: [PATCH] move checkpoint message to variable message --- paddle/fluid/operators/detail/grpc_client.cc | 9 +-------- paddle/fluid/operators/listen_and_serv_op.cc | 3 ++- 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/paddle/fluid/operators/detail/grpc_client.cc b/paddle/fluid/operators/detail/grpc_client.cc index 17476ab51..7a63e39d5 100644 --- a/paddle/fluid/operators/detail/grpc_client.cc +++ b/paddle/fluid/operators/detail/grpc_client.cc @@ -236,17 +236,10 @@ void GRPCClient::AsyncCheckpointNotify(const std::string& ep, CheckpointNotifyProcessor* s = new CheckpointNotifyProcessor(ch); s->Prepare(time_out); - s->response_call_back_ = nullptr; sendrecv::VariableMessage req; req.set_varname(CHECKPOINT_SAVE_MESSAGE); - req.out_varname(dir); - - auto call = s->stub_g_.PrepareUnaryCall( - s->context_.get(), "/sendrecv.SendRecvService/CheckpointNotify", req, - &cq_); - call->StartCall(); - call->Finish(&s->reply_, &s->status_, reinterpret_cast(s)); + req.set_out_varname(dir); auto rpc = s->stub_->AsyncCheckpointNotify(s->context_.get(), req, &cq_); rpc->Finish(&s->reply_, &s->status_, reinterpret_cast(s)); diff --git a/paddle/fluid/operators/listen_and_serv_op.cc b/paddle/fluid/operators/listen_and_serv_op.cc index 698ff2299..7294acc3e 100644 --- a/paddle/fluid/operators/listen_and_serv_op.cc +++ b/paddle/fluid/operators/listen_and_serv_op.cc @@ -329,7 +329,8 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope, // Write to a file of server selected port for python use. SavePort(); if (sync_mode) { - RunSyncLoop(&executor, program, &recv_scope, prefetch_block_id_list); + RunSyncLoop(&executor, program, &recv_scope, prefetch_block_id_list, + checkpoint_point_block_id); } else { RunAsyncLoop(&executor, program); } -- GitLab