Struct SimulatedBackend 

pub struct SimulatedBackend { /* private fields */ }

In-process backend using std::sync::mpsc channels.

Designed for testing collectives and DDP without spawning processes. Create all ranks via SimulatedBackend::create_group, which returns one SimulatedBackend per rank.

Implementations

impl SimulatedBackend

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

Create a group of world_size simulated backends, one per rank.

Returns a Vec<SimulatedBackend> where index i is rank i.

Trait Implementations

impl Backend for SimulatedBackend

fn rank(&self) -> usize

This rank’s index in the world (0-based).

fn world_size(&self) -> usize

Total number of ranks in the process group.

fn send(&self, data: &[u8], dst_rank: usize) -> FerrotorchResult<()>

Send data to dst_rank.

fn recv(&self, dst: &mut [u8], src_rank: usize) -> FerrotorchResult<()>

Receive into dst from src_rank. The caller must allocate dst with the correct length before calling.

fn recv_timeout( &self, dst: &mut [u8], src_rank: usize, timeout: Duration, ) -> FerrotorchResult<()>

Receive into dst from src_rank with a timeout.

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

Block until every rank has reached this barrier.