Enum JitOptimization

pub enum JitOptimization<R: JitRuntime> {
    ElementWise(ElemwiseOptimization<R>),
    Matmul(MatmulOptimization<R>),
}

Available on crate feature fusion only.

Fusion optimization type for JIT.

More optimization variants should be added here.

Variants

ElementWise(ElemwiseOptimization<R>)

Element wise optimization.

Matmul(MatmulOptimization<R>)

Matrix multiplication optimization.

Trait Implementations

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.