diff --git a/paddle/fluid/framework/fleet/fleet_wrapper.h b/paddle/fluid/framework/fleet/fleet_wrapper.h index 29436772214849347ed632b2c537104f0e6769a3..386e711ff71dbf978cbcb620589490d3f06d3c53 100644 --- a/paddle/fluid/framework/fleet/fleet_wrapper.h +++ b/paddle/fluid/framework/fleet/fleet_wrapper.h @@ -121,12 +121,17 @@ class FleetWrapper { void StopServer(); uint64_t RunServer(); void GatherServers(const std::vector& host_sign_list, int node_num); + // gather client ip void GatherClients(const std::vector& host_sign_list); + // get client info std::vector GetClientsInfo(); + // create client to client connection void CreateClient2ClientConnection(); + // register client to client communication typedef std::function MsgHandlerFunc; int RegisterClientToClientMsgHandler(int msg_type, MsgHandlerFunc handler); + // send client to client message std::future SendClientToClientMsg(int msg_type, int to_client_id, const std::string& msg);