comm_utils.h File Reference

#include <raft/core/comms.hpp>
#include <raft/util/cuda_utils.cuh>

Include dependency graph for comm_utils.h:

Go to the source code of this file.

Namespaces
	MLCommon
	MLCommon::opg

Functions
template<typename T >
void	MLCommon::opg::reduce_single_sum (T *out, const T *in, const raft::comms::comms_t &comm, cudaStream_t stream, int root=0)
template<typename T >
void	MLCommon::opg::allreduce_single_sum (T *out, const T *in, const raft::comms::comms_t &comm, cudaStream_t stream)