Struct CostModel 

pub struct CostModel { /* private fields */ }

The main cost model; use CostModel::estimate_graph to get a full GraphCostSummary.

Implementations

impl CostModel

pub fn new(config: CostModelConfig) -> Self

Create with the supplied config.

pub fn with_default() -> Self

Create with default config (f64 elements, no throughput hint).

pub fn estimate_graph(&self, graph: &EinsumGraph) -> GraphCostSummary

Estimate costs for the entire graph, returning a GraphCostSummary.

pub fn estimate_node( &self, node: &EinsumNode, input_shapes: &[Vec<usize>], ) -> NodeCostEstimate

Estimate cost for a single node given known input shapes.

pub fn estimate_einsum_flops( equation: &str, input_shapes: &[Vec<usize>], ) -> FlopEstimate

Estimate FLOPs for an einsum contraction.

Strategy: multiply all unique index dimension sizes together. That product is the number of multiply-add operations.

pub fn infer_output_shape( node: &EinsumNode, input_shapes: &[Vec<usize>], ) -> Vec<usize>

Infer the output shape for a node given input shapes.

For einsum, parses the equation to determine output dimension sizes. For other ops, attempts shape propagation heuristics. Falls back to a non-empty placeholder [1] when inference is not possible.

pub fn rank_by_flops(summary: &GraphCostSummary) -> Vec<&NodeCostEstimate>

Sort nodes by descending FLOP cost.