Struct PipelineConfig 

pub struct PipelineConfig {
    pub num_stages: usize,
    pub num_micro_batches: usize,
}

Configuration of a pipeline: number of stages and micro-batches.

Fields

num_stages: usize

Number of pipeline stages P (pipeline depth).

num_micro_batches: usize

Number of micro-batches M the minibatch is split into.

Implementations

impl PipelineConfig

pub fn new(num_stages: usize, num_micro_batches: usize) -> Result<Self>

Create a configuration, validating that both counts are at least one.

pub fn analytical_bubble_fraction(&self) -> f64

Analytical GPipe bubble fraction (P-1)/(M+P-1).

This is the fraction of stage-time wasted to pipeline fill/drain under the idealised uniform-cost model. The generated schedule’s measured idle fraction equals this value when all stage costs are uniform.

pub fn analytical_utilization(&self) -> f64

Analytical pipeline utilisation M/(M+P-1) = 1 - bubble_fraction.

Trait Implementations

impl Clone for PipelineConfig

fn clone(&self) -> PipelineConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for PipelineConfig

impl Debug for PipelineConfig

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

Formats the value using the given formatter.

impl Eq for PipelineConfig

impl PartialEq for PipelineConfig

fn eq(&self, other: &PipelineConfig) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for PipelineConfig