Struct MockDistributedCommunication 

pub struct MockDistributedCommunication { /* private fields */ }

A mock implementation of distributed communication for testing.

Implementations

impl MockDistributedCommunication

pub fn new(numworkers: usize, rank: usize) -> Self

Create a new mock distributed communication channel.

Trait Implementations

impl DistributedCommunication for MockDistributedCommunication

fn send( &self, _tensor: Box<dyn ArrayProtocol>, destination: usize, ) -> CoreResult<()>

Send a tensor to another worker.

fn recv(&self, source: usize) -> CoreResult<Box<dyn ArrayProtocol>>

Receive a tensor from another worker.

fn broadcast( &self, tensor: Box<dyn ArrayProtocol>, ) -> CoreResult<Box<dyn ArrayProtocol>>

Broadcast a tensor from the master to all workers.

fn gather( &self, tensor: Box<dyn ArrayProtocol>, ) -> CoreResult<Vec<Box<dyn ArrayProtocol>>>

Gather tensors from all workers to the master.

fn scatter( &self, tensors: Vec<Box<dyn ArrayProtocol>>, ) -> CoreResult<Box<dyn ArrayProtocol>>

Scatter tensors from the master to all workers.

fn reduce( &self, tensor: Box<dyn ArrayProtocol>, op: &str, ) -> CoreResult<Box<dyn ArrayProtocol>>

Reduce tensors from all workers to the master.

fn all_reduce( &self, tensor: Box<dyn ArrayProtocol>, op: &str, ) -> CoreResult<Box<dyn ArrayProtocol>>

All-reduce tensors across all workers.

fn all_gather( &self, tensor: Box<dyn ArrayProtocol>, ) -> CoreResult<Vec<Box<dyn ArrayProtocol>>>

All-gather tensors from all workers to all workers.

fn barrier(&self) -> CoreResult<()>

Barrier synchronization.

fn box_clone(&self) -> Box<dyn DistributedCommunication>

Clone this communication channel.