Struct GraphJITCompiler 

pub struct GraphJITCompiler {
    pub backends: Vec<JITBackend>,
    pub default_opt_level: OptimizationLevel,
    pub kernel_cache: HashMap<String, CompiledKernel>,
    pub stats: CompilationStats,
    pub fusion_rules: Vec<FusionRule>,
}

JIT compiler for graph operations

Fields

backends: Vec<JITBackend>

Available backends

default_opt_level: OptimizationLevel

Default optimization level

kernel_cache: HashMap<String, CompiledKernel>

Compiled kernel cache

stats: CompilationStats

Compilation statistics

fusion_rules: Vec<FusionRule>

Kernel fusion rules

Implementations

impl GraphJITCompiler

pub fn new() -> Self

Create a new JIT compiler

pub fn add_backend(&mut self, backend: JITBackend)

Add a backend to the compiler

pub fn compile_operation( &mut self, operation: GraphOperation, input_shapes: &[Vec<usize>], backend: Option<JITBackend>, ) -> Result<CompiledKernel, JITError>

Compile a graph operation to optimized kernel

pub fn execute_kernel( &self, kernel: &CompiledKernel, inputs: &[Tensor], ) -> Result<Vec<Tensor>, JITError>

Execute a compiled kernel with given inputs

pub fn fuse_operations( &mut self, operations: &[GraphOperation], input_shapes: &[Vec<usize>], ) -> Result<CompiledKernel, JITError>

Analyze and fuse multiple operations for better performance

pub fn get_stats(&self) -> &CompilationStats

Get compilation statistics

pub fn clear_cache(&mut self)

Clear the kernel cache

Trait Implementations

impl Debug for GraphJITCompiler

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

Formats the value using the given formatter.

impl Default for GraphJITCompiler

fn default() -> Self

Returns the “default value” for a type.