cpp/include/raft/comms/helper.hpp
void build_comms_nccl_only(handle_t* handle, ncclComm_t nccl_comm, int num_ranks, int rank)
{
  cudaStream_t stream = handle->get_stream();

  auto communicator = std::make_shared<comms_t>(
    std::unique_ptr<comms_iface>(new raft::comms::std_comms(nccl_comm, num_ranks, rank, stream)));
  handle->set_comms(communicator);
}
/**
 * Function to construct comms_t and inject it on a handle_t. This
 * is used for convenience in the Python layer.
 *
 * @param handle raft::handle_t for injecting the comms
 * @param nccl_comm initialized NCCL communicator to use for collectives
 * @param num_ranks number of ranks in communicator clique
 * @param rank rank of local instance
 */
Function to construct comms_t and inject it on a handle_t. This
is used for convenience in the Python layer.

@param handle raft::handle_t for injecting the comms
@param nccl_comm initialized NCCL communicator to use for collectives
@param num_ranks number of ranks in communicator clique
@param rank rank of local instance
Function to construct comms_t and inject it on a handle_t . This is used for convenience in the Python layer . @param handle raft :: handle_t for injecting the comms @param nccl_comm initialized NCCL communicator to use for collectives @param num_ranks number of ranks in communicator clique @param rank rank of local instance
void build_comms_nccl_only(handle_t* handle, ncclComm_t nccl_comm, int num_ranks, int rank)
{
  cudaStream_t stream = handle->get_stream();
  auto communicator = std::make_shared<comms_t>(
    std::unique_ptr<comms_iface>(new raft::comms::std_comms(nccl_comm, num_ranks, rank, stream)));
  handle->set_comms(communicator);
}
void build_comms_nccl_only ( handle_t * handle , ncclComm_t nccl_comm , int num_ranks , int rank ) { cudaStream_t stream = handle -> get_stream ( ) ; auto communicator = std :: make_shared < comms_t > ( std :: unique_ptr < comms_iface > ( new raft :: comms :: std_comms ( nccl_comm , num_ranks , rank , stream ) ) ) ; handle -> set_comms ( communicator ) ; }
Function to construct comms_t and inject it on a handle_t.
Function to construct comms_t and inject it on a handle_t .