diff --git a/paddle/operators/listen_and_serv_op.cc b/paddle/operators/listen_and_serv_op.cc index 5745938ed9b31e596acac844fa1b16793340838b..099f6b23736adcc2a6e9c27dca297178687ae785 100644 --- a/paddle/operators/listen_and_serv_op.cc +++ b/paddle/operators/listen_and_serv_op.cc @@ -204,4 +204,4 @@ from send_op and send back variables to recv_op. namespace ops = paddle::operators; REGISTER_OPERATOR(listen_and_serv, ops::ListenAndServOp, - ops::ListenAndServOpMaker); \ No newline at end of file + ops::ListenAndServOpMaker); diff --git a/paddle/operators/send_op.cc b/paddle/operators/send_op.cc index 291d19ba8bd6113fb69719d5385792ad5cde387d..ee0f268b0e4dfa23bf878d71404d47553183a977 100644 --- a/paddle/operators/send_op.cc +++ b/paddle/operators/send_op.cc @@ -65,9 +65,9 @@ class SendOp : public framework::OperatorBase { if (outs.size() > 0) { for (size_t i = 0; i < outs.size(); i++) { VLOG(3) << "getting " << outs[i] << " from " << epmap[i]; - client_.AsyncGetVariable(epmap[i], ctx, scope, outs[i]); + rpc_client->AsyncGetVariable(epmap[i], ctx, scope, outs[i]); } - PADDLE_ENFORCE(client_.Wait()); + PADDLE_ENFORCE(rpc_client->Wait()); } } };