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.