docs/api/device__comm_8h_source.html

#ifndef DEVICE_COMM_H

#define DEVICE_COMM_H


#include <Kokkos_Core.hpp>


#ifdef Mallard_HAS_NCCL

#include <nccl.h>

#endif


namespace comm {


constexpr bool nccl_available() {

#ifdef Mallard_HAS_NCCL

    return true;

#else

    return false;

#endif

}


#ifdef Mallard_HAS_NCCL

ncclComm_t nccl();


void finalize_nccl();

#endif


void enqueue_wait(const Kokkos::DefaultExecutionSpace & waiter, const Kokkos::DefaultExecutionSpace & done);


} // namespace comm


#endif // DEVICE_COMM_H

dot
KOKKOS_INLINE_FUNCTION rtype dot(const rtype *a, const rtype *b)
Compute the dot product of two arrays.
Definition common_math.h:32

comm
Definition comm.cpp:24

comm::nccl_available
constexpr bool nccl_available()
Whether this build can exchange device buffers with NCCL.
Definition device_comm.h:25

comm::enqueue_wait
void enqueue_wait(const Kokkos::DefaultExecutionSpace &waiter, const Kokkos::DefaultExecutionSpace &done)
Make the work enqueued later on waiter wait for the work enqueued so far on done, without blocking th...
Definition device_comm.cpp:54