Struct ParameterServer 

pub struct ParameterServer { /* private fields */ }

Parameter server supporting BSP, ASP, and SSP consistency models

Implementations

impl ParameterServer

pub fn new(config: ParamServerConfig) -> Self

Create a new parameter server with the given configuration

pub fn register_worker(&mut self) -> usize

Register a new worker and return its ID

pub fn init_parameter(&mut self, key: impl Into<String>, values: Vec<f64>)

Initialize a parameter with the given key and values

pub fn push(&mut self, update: ParameterUpdate) -> CoreResult<()>

Push an update from a worker

Behavior depends on the consistency model:

• BSP: buffers until all workers push, then applies via barrier_sync
• ASP: applies immediately
• SSP: applies if within staleness bound, otherwise buffers

pub fn pull(&self, key: &str, worker_id: usize) -> CoreResult<(Vec<f64>, u64)>

Pull current parameter values for a given key

pub fn aggregate_updates( updates: &[ParameterUpdate], method: &AggregationMethod, ) -> Vec<f64>

Aggregate updates using the configured method

pub fn barrier_sync(&mut self) -> CoreResult<()>

BSP barrier synchronization

Applies all buffered updates and increments the global version. Returns an error if not all alive workers have pushed.

pub fn checkpoint(&self) -> ServerCheckpoint

Create a checkpoint of the current server state

pub fn restore( checkpoint: &ServerCheckpoint, config: ParamServerConfig, ) -> CoreResult<Self>

Restore a parameter server from a checkpoint

pub fn global_version(&self) -> u64

Get the current global version

pub fn num_workers(&self) -> usize

Get number of registered workers

pub fn workers(&self) -> &[WorkerState]

Get a reference to the worker states

pub fn workers_mut(&mut self) -> &mut Vec<WorkerState>

Get a mutable reference to the worker states

pub fn config(&self) -> &ParamServerConfig

Get the configuration

Trait Implementations

impl Debug for ParameterServer

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

Formats the value using the given formatter.