Struct FusionJitRuntime

pub struct FusionJitRuntime<R: JitRuntime, BT: BoolElement> { /* private fields */ }

Available on crate feature fusion only.

Fusion runtime for JIT runtimes.

Trait Implementations

impl<R: Debug + JitRuntime, BT: Debug + BoolElement> Debug for FusionJitRuntime<R, BT>

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

Formats the value using the given formatter.

impl<R: JitRuntime, BT: BoolElement> FusionRuntime for FusionJitRuntime<R, BT>

type OptimizationState = JitOptimizationState

The state that can be serialized for an optimization.

type Optimization = JitOptimization<R>

Optimization type for the backend.

type FusionHandle = JitFusionHandle<R>

Handle used to store tensor dynamically.

type FusionDevice = <R as JitRuntime>::JitDevice

Device used by the runtime.

type FusionClient = MutexFusionClient<FusionJitRuntime<R, BT>>

The client to interact with the runtime.

type BoolRepr = BT

The type that represents booleans on the backend.

fn optimizations( device: R::Device, ) -> Vec<Box<dyn OptimizationBuilder<Self::Optimization>>>

The list of optimizations that will be used to optimize the computational graph.

impl<R, BT> Optimization<FusionJitRuntime<R, BT>> for JitOptimization<R>

where R: JitRuntime, BT: BoolElement,

fn execute(&mut self, context: &mut Context<'_, JitFusionHandle<R>>)

Execute the operation.

fn len(&self) -> usize

The number of registered operations in this optimization.

fn to_state(&self) -> JitOptimizationState

Returns the state that can be serialized.

fn from_state(device: &R::Device, state: JitOptimizationState) -> Self

Create the optimization from the state.

fn is_empty(&self) -> bool

If the current optimization is empty.