Fix compilie error

emailweixu-patch-1
Yancey1989 8 years ago
parent 82e41ce38f
commit d762c6d7ff

@ -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);
ops::ListenAndServOpMaker);

@ -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());
}
}
};

Loading…
Cancel
Save