Enum PipelineSchedule 

pub enum PipelineSchedule {
    GPipe,
    OneForwardOneBackward,
}

Which pipeline schedule to generate.

Variants

GPipe

All-forward-then-all-backward (GPipe). Simple, but stashes M activations per stage.

OneForwardOneBackward

One-forward-one-backward steady state (PipeDream-Flush). Same bubble as GPipe but bounds stashed activations to the pipeline depth.

Implementations

impl PipelineSchedule

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

Human-readable name of the schedule.

Trait Implementations

impl Clone for PipelineSchedule

fn clone(&self) -> PipelineSchedule

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for PipelineSchedule

impl Debug for PipelineSchedule

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

Formats the value using the given formatter.

impl Eq for PipelineSchedule

impl PartialEq for PipelineSchedule

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for PipelineSchedule