Expand description
Graph compilation and caching infrastructure.
This module provides ahead-of-time graph optimization and compilation capabilities:
CompiledGraph: Optimized, executable representation of computation graphsGraphCompiler: Applies optimization passes and produces compiled graphsCompilationCache: Caches compiled graphs to avoid recompilationTlCompilableExecutor: Trait for executors that support graph compilation
§Example
use tensorlogic_infer::compilation::{GraphCompiler, CompilationConfig, OptimizationLevel};
use tensorlogic_infer::DummyExecutor;
use tensorlogic_ir::EinsumGraph;
let mut compiler = GraphCompiler::new(CompilationConfig {
optimization_level: OptimizationLevel::Aggressive,
..Default::default()
});
let graph = EinsumGraph::new();
let compiled = compiler.compile(&graph).unwrap();Structs§
- Cache
Stats - Statistics for the compilation cache.
- Compilation
Cache - Cache for compiled graphs.
- Compilation
Config - Configuration for graph compilation.
- Compilation
Key - Cache key for compiled graphs.
- Compilation
Stats - Statistics about the compilation process.
- Compiled
Graph - Compiled representation of a computation graph.
- Graph
Compiler - Graph compiler that applies optimization passes.
Enums§
- Optimization
Level - Optimization level for graph compilation.
Traits§
- TlCompilable
Executor - Trait for executors that support graph compilation.