Collection Wrappers#

template<typename T>
rmm::device_uvector<T> device_allgatherv(raft::handle_t const &handle, raft::comms::comms_t const &comm, raft::device_span<T const> d_input)#

Gather the span of data from all ranks and broadcast the combined data to all ranks.

Parameters:
  • handle[in] RAFT handle object to encapsulate resources (e.g. CUDA stream, communicator, and handles to various CUDA libraries) to run graph algorithms.

  • comm[in] Raft comms that manages underlying NCCL comms handles across the ranks.

  • d_input[in] The span of data to perform the ‘allgatherv’.

Returns:

A vector containing the combined data of all ranks.