Struct ModelCompilePipeline 

pub struct ModelCompilePipeline { /* private fields */ }

Compile-once / specialize-per-variant pipeline with optional FIFO cache.

Implementations

impl ModelCompilePipeline

pub fn new(device: Device) -> Self

pub fn with_capacity(device: Device, capacity: usize) -> Self

pub fn device(&self) -> Device

pub fn has_template(&self) -> bool

pub fn build_template<F>( &mut self, build_hir: F, options: &CompileOptions, ) -> Result<&CompileResult, LowerError>

where F: FnOnce() -> HirModule,

Step 1 — run fusion pipeline on symbolic HIR (dynamic dims allowed).

pub fn template_binding_manifest(&self) -> BindingManifest

pub fn specialize_template( &self, binding: &DimBinding, options: &CompileOptions, ) -> CompileResult

Step 2 — bind symbolic dims and replan buffers.

pub fn compile_lir( &self, specialized: CompileResult, options: &CompileOptions, ) -> CompiledGraph

Step 3 — backend executable from specialized LIR.

pub fn get_or_compile<F>( &mut self, key: u64, binding: &DimBinding, build_hir: F, options: &CompileOptions, ) -> Result<&mut CompiledGraph, LowerError>

where F: FnOnce() -> HirModule,

Full pipeline: template (once) → specialize → compile; cached by key.

pub fn binding_manifest_for_binding( &self, binding: &DimBinding, options: &CompileOptions, ) -> BindingManifest

Manifest for a variant without storing specialized LIR in the cache.

pub fn binding_manifest_for_component( &self, component: &ModelComponent, options: &CompileOptions, ) -> BindingManifest

Layout for a full ModelComponent (specialized parameter block).

pub fn get_or_compile_component<F>( &mut self, component: &ModelComponent, build_hir: F, options: &CompileOptions, ) -> Result<(&mut CompiledGraph, BindingManifest), LowerError>

where F: FnOnce() -> HirModule,

Template → specialize → compile; keyed by ModelComponent::cache_key.

pub fn contains(&self, key: u64) -> bool

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn template_result(&self) -> Option<&CompileResult>

Symbolic template from Self::build_template / Self::get_or_compile.

pub fn ensure_template<F: FnOnce() -> HirModule>( &mut self, build_hir: F, options: &CompileOptions, ) -> Result<&CompileResult, LowerError>

Build the symbolic template once (no specialization).

pub fn get_or_specialize_aot<F: FnOnce() -> HirModule>( &mut self, aot: &AotCache, disk_base: &str, key: u64, binding: &DimBinding, build_hir: F, options: &CompileOptions, ) -> Result<&mut CompiledGraph, AotCacheError>

Disk-backed specialize ([CompilationMode::Aot]); caches by key.