Struct CompiledBackend

pub struct CompiledBackend<C: Compiler> { /* private fields */ }

Compiled backend that holds compiler, buffers and programs

Implementations

impl<C: Compiler> CompiledBackend<C>

pub fn new(compiler: C) -> Self

Initialize new compiled backend using provided compiler

Trait Implementations

impl<C: Compiler> RuntimeBackend for CompiledBackend<C>

fn is_evaluated(&self, x: Id) -> bool

Is tensor x evaluated?

fn is_free_id(&self, x: Id) -> bool

Check if there are no more buffers on id x

fn remove(&mut self, x: Id) -> Result<(), ZyxError>

Delete all memory used by tensor x.

fn store<T: Scalar, IT>(&mut self, x: Id, iter: IT) -> Result<(), ZyxError>

where IT: IntoIterator<Item = T>, IT::IntoIter: ExactSizeIterator,

Store iterator into runtime backend

fn load<T: Scalar>(&mut self, x: Id, numel: usize) -> Result<Vec<T>, ZyxError>

Load evaluated tensor x.

fn evaluate( &mut self, rcs: BTreeMap<Id, u32>, order: &[Id], nodes: &[Node], ) -> Result<(), ZyxError>

Evaluate tensors to_eval with given graph of nodes and recommended order of evaluation.