Struct FusionGraph 

pub struct FusionGraph { /* private fields */ }

An append-only builder for an operation DAG.

Implementations

impl FusionGraph

pub fn new() -> Self

Creates an empty graph.

pub fn add_op( &mut self, kind: OpKind, inputs: Vec<usize>, output_shape: Vec<usize>, dtype_bytes: usize, ) -> usize

Appends an op and returns its freshly assigned id.

inputs must reference the ids of previously added ops (an empty list denotes a source/leaf). Validity and acyclicity are checked by FusionGraph::validate, not here, so a forward reference can be built and later rejected.

pub fn ops(&self) -> &[FusionOp]

Returns the ops in insertion order.

pub fn num_ops(&self) -> usize

Number of ops in the graph.

pub fn is_empty(&self) -> bool

Returns true if the graph has no ops.

pub fn validate(&self) -> Result<(), GpuOptimError>

Validates the graph: every input must reference an existing op, no op may reference itself, dtype_bytes must be non-zero, byte counts must not overflow, and the graph must be acyclic.

Returns GpuOptimError::InvalidState for a dangling input, a self-reference, or a cycle.

Trait Implementations

impl Clone for FusionGraph

fn clone(&self) -> FusionGraph

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FusionGraph

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

Formats the value using the given formatter.

impl Default for FusionGraph

fn default() -> FusionGraph

Returns the “default value” for a type.