集合包装器#

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 组合数据的向量。