Struct Communicator 

pub struct Communicator<C> { /* private fields */ }

Implementations

impl<C> Communicator<C>

pub fn new<K, E>( contexts: Vec<C>, initialize: impl FnOnce(&C) -> Result<K, E>, ) -> Result<(Self, K), E>

where E: From<InProcessError>,

pub fn world_size(&self) -> usize

pub const fn transport(&self) -> CollectiveTransport

pub fn collective<T, D, E>(&self) -> InProcessCollective<'_, T, D, E>

where D: InProcessDevice<T, Context = C>, E: From<D::Error> + From<InProcessError>,

pub fn barrier(&self) -> Result<CollectiveStats, InProcessError>

impl<B: Backend> Communicator<TensorDevice<B>>

pub fn tensor_collective<T: TensorElement>( &self, ) -> InProcessCollective<'_, T, TensorDevice<B>, TensorDeviceError>

Trait Implementations

impl<C: Clone> Clone for Communicator<C>

fn clone(&self) -> Communicator<C>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<C: Debug> Debug for Communicator<C>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.