Struct HPGO::orchestration::orchestrate_hierarchical::HierarchicalConductor

pub struct HierarchicalConductor {
    pub ctx: Matrix,
    pub m: Model,
}

Fields

ctx: Matrix

m: Model

Methods

impl HierarchicalConductor

pub fn new_from_torch_graph(
    filename: &str,
    pbs: u32,
    gbs: u32
) -> HierarchicalConductor

pub fn compute_plan_hierarchical(
    &mut self,
    num_machines: u32,
    num_cards_per_machine: u32,
    bandwidth: f64,
    final_level: bool
)

Trait Implementations

impl Debug for HierarchicalConductor

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

Formats the value using the given formatter.

impl Orchestrate for HierarchicalConductor

fn orchestrate(&mut self)

fn compute_plan(&mut self)

fn analyse_plan(&self)

fn return_plan(&self) -> Box<dyn OrchestrationResult>