|
void | Init (const grape::CommSpec &comm_spec) |
| Initialize message manager.
|
|
void | InitBuffer (size_t send_buffer_capacity, size_t recv_buffer_capacity) |
|
void | DropBuffer () |
|
void | Start () |
| This function will be called before Init step of applications.
|
|
void | StartARound () |
| This function will be called before each evaluation step of applications.
|
|
void | FinishARound () |
| This function will be called after each evaluation step of applications.
|
|
void | Finalize () const |
| This function will be called after the evaluation of applications.
|
|
bool | ToTerminate () const |
| This function will be called by worker after a step to determine whether evaluation is terminated.
|
|
size_t | GetMsgSize () const |
| Get size of messages sent by this message manager instance. The return value is valid only after FinishARound is called. StartARound will reset the value to zero.
|
|
void | ForceContinue () |
| Force continue to evaluate one more round even if all workers stop sending message.
|
|
double | GetAccumulatedCommTime () const |
|
dev::MessageManager | DeviceObject () |
|
void | DeviceWarmup (size_t np) |
| For some GPU servers, the first kernel always takes a long time. This is a Dummy function to warm up the device.
|
|
template<typename GRAPH_T , typename MESSAGE_T = grape::EmptyType, typename FUNC_T > |
void | ParallelProcess (const GRAPH_T &frag, FUNC_T func) |
|
template<typename MESSAGE_T = grape::EmptyType, typename FUNC_T > |
void | ParallelProcess (FUNC_T func) |
|
Stream & | stream () |
|
ncclComm_t | nccl_comm () |
|