Struct GraphScheduler 

pub struct GraphScheduler { /* private fields */ }

Advanced scheduler for computation graphs

Implementations

impl GraphScheduler

pub fn new() -> Self

Create a new scheduler

pub fn set_operation_cost(&mut self, node_idx: usize, cost: f64)

Set the cost for an operation

pub fn set_tensor_memory(&mut self, tensor_idx: usize, size: usize)

Set memory size for a tensor

pub fn schedule( &self, graph: &EinsumGraph, objective: SchedulingObjective, ) -> Result<ExecutionSchedule, IrError>

Generate a schedule optimized for the given objective

Trait Implementations

impl Default for GraphScheduler

fn default() -> Self

Returns the “default value” for a type.