Struct MockComm 

pub struct MockComm {
    pub rank: usize,
    pub world_size: usize,
    /* private fields */
}

Mock communication backend for testing distributed operations.

This implementation simulates collective operations in-memory, allowing testing of distributed code paths on a single machine.

Fields

rank: usize

This rank’s ID in the mesh

world_size: usize

Total number of ranks

Implementations

impl MockComm

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

Create a new MockComm for a specific rank

pub fn with_shared_state(rank: usize, state: Arc<RwLock<MockCommState>>) -> Self

Create a new MockComm with shared state (for multi-rank simulation)

pub fn create_group(world_size: usize) -> Vec<Self>

Create a group of MockComm instances that share state

Trait Implementations

impl Debug for MockComm

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

Formats the value using the given formatter.

impl MeshComm for MockComm

fn all_reduce( &self, buf: &mut [u8], dtype: DType, op: ReduceOp, _group: &str, ) -> Result<(), String>

All-reduce operation: reduce values across all ranks in a group

fn all_gather( &self, local: &[u8], out: &mut [u8], _dtype: DType, _group: &str, ) -> Result<(), String>

All-gather: gather data from all ranks

fn broadcast( &self, buf: &mut [u8], root_rank: usize, _group: &str, ) -> Result<(), String>

Broadcast from root rank to all ranks in group

fn reduce_scatter( &self, buf: &mut [u8], out: &mut [u8], op: ReduceOp, group: &str, ) -> Result<(), String>

Reduce-scatter: reduce and distribute results

fn barrier(&self, _group: &str) -> Result<(), String>

Barrier synchronization

fn send(&self, buf: &[u8], dest_rank: usize) -> Result<(), String>

Send data to a specific rank

fn recv(&self, buf: &mut [u8], src_rank: usize) -> Result<(), String>

Receive data from a specific rank