Struct MockBackend 

pub struct MockBackend { /* private fields */ }

A mock backend for testing distributed operations in a single process. Simulates distributed communication without actual network operations.

Implementations

impl MockBackend

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

Creates a collection of mock backends for testing.

pub fn single() -> Self

Creates a single mock backend (rank 0, world size 1).

Trait Implementations

impl Backend for MockBackend

fn name(&self) -> &'static str

Returns the name of the backend.

fn rank(&self) -> usize

Returns the rank of this process.

fn world_size(&self) -> usize

Returns the total world size.

fn all_reduce(&self, data: &mut [f32], op: ReduceOp)

Performs all-reduce operation.

fn broadcast(&self, data: &mut [f32], src: usize)

Broadcasts data from a source rank.

fn all_gather(&self, send_data: &[f32], recv_data: &mut [f32])

Performs all-gather operation.

fn reduce_scatter(&self, send_data: &[f32], recv_data: &mut [f32], op: ReduceOp)

Performs reduce-scatter operation.

fn gather(&self, send_data: &[f32], recv_data: &mut [f32], dst: usize)

Performs gather operation.

fn scatter(&self, send_data: &[f32], recv_data: &mut [f32], src: usize)

Performs scatter operation.

fn reduce( &self, send_data: &[f32], recv_data: &mut [f32], dst: usize, op: ReduceOp, )

Performs reduce operation (result only on dst rank).

fn barrier(&self)

Synchronizes all processes.

fn send(&self, data: &[f32], dst: usize, tag: usize)

Sends data to a specific rank.

fn recv(&self, data: &mut [f32], src: usize, tag: usize)

Receives data from a specific rank.