集合包装器#
- 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)#从所有 rank 收集数据跨度,并将组合数据广播到所有 rank。
- 参数:
handle – [in] RAFT 句柄对象,用于封装运行图算法所需的资源(例如 CUDA 流、通信器以及各种 CUDA 库的句柄)。
comm – [in] Raft 通信器,管理跨 rank 的底层 NCCL 通信句柄。
d_input – [in] 要执行 'allgatherv' 操作的数据跨度。
- 返回值:
包含所有 rank 组合数据的向量。