|
void | InitCommunicator (MPI_Comm comm, ncclComm_t nccl_comm) |
|
template<typename T > |
void | Sum (T msg_in, T &msg_out, const Stream &stream) |
|
template<typename T > |
void | Sum (T *msg_in, T &msg_out, const Stream &stream) |
|
template<typename T > |
void | Sum (T msg_in, T &msg_out) |
|
template<typename T > |
void | Min (T msg_in, T &msg_out, const Stream &stream) |
|
template<typename T > |
void | Min (T *msg_in, T &msg_out, const Stream &stream) |
|
template<typename T > |
void | Min (T msg_in, T &msg_out) |
|
template<typename T > |
void | Max (T msg_in, T &msg_out, const Stream &stream) |
|
template<typename T > |
void | Max (T *msg_in, T &msg_out, const Stream &stream) |
|
template<typename T > |
void | Max (T msg_in, T &msg_out) |
|
template<typename T > |
std::vector< T > | AllGather (T msg_in) |
|
|
template<typename T > |
ncclDataType_t | ncclType () |
|
template<typename T > |
MPI_Datatype | mpiType () |
|
template<typename T > |
void | reduce (T *msg_in, T *msg_out, ncclRedOp_t op, const Stream &stream) |
|
template<typename T > |
void | reduce (T msg_in, T &msg_out, ncclRedOp_t op, const Stream &stream) |
|
template<typename T > |
void | reduce (T *msg_in, T &msg_out, ncclRedOp_t op, const Stream &stream) |
|
template<typename T > |
void | reduce (const T *msg_in, T *msg_out, MPI_Op op) |
|