Struct CommunicationManager 

pub struct CommunicationManager { /* private fields */ }

Communication manager for distributed operations

Implementations

impl CommunicationManager

pub fn new(config: &DistributedConfig) -> Result<Self, DistributedError>

pub fn rank(&self) -> usize

Get the rank of this worker

pub fn num_workers(&self) -> usize

Get the number of workers

pub fn all_reduce(&mut self, data: &[f32]) -> Result<Vec<f32>, DistributedError>

pub fn all_reduce_sum( &mut self, data: &[f32], ) -> Result<Vec<f32>, DistributedError>

pub fn send_to( &mut self, _target_rank: usize, _data: &[Tensor], ) -> Result<(), DistributedError>

pub fn receive_from( &mut self, _source_rank: usize, ) -> Result<Vec<Tensor>, DistributedError>

pub fn request_boundary_features( &mut self, _target_worker: usize, ) -> Result<Tensor, DistributedError>

Trait Implementations

impl Debug for CommunicationManager

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

Formats the value using the given formatter.