Skip to main content

apply_graph_optimizations

Function apply_graph_optimizations 

Source
pub fn apply_graph_optimizations(
    graph: &EinsumGraph,
    config: &GraphOptConfig,
) -> Result<(EinsumGraph, GraphOptStats)>
Expand description

Apply integrated graph optimizations to a compiled EinsumGraph

This is the main entry point for applying IR-level optimizations to graphs produced by the compiler. It orchestrates multiple optimization passes in an intelligent order.

ยงExample

use tensorlogic_compiler::passes::graph_opt_integration::{
    apply_graph_optimizations, GraphOptConfig
};
use tensorlogic_ir::EinsumGraph;

let mut graph = EinsumGraph::new();
// ... compile logic expressions into graph ...

let config = GraphOptConfig::default();
let (optimized, stats) = apply_graph_optimizations(&graph, &config).unwrap();

println!("Applied {} optimizations", stats.total_optimizations());
println!("Estimated speedup: {:.2}x", stats.estimated_speedup);