Struct ElemwiseOptimization

pub struct ElemwiseOptimization<R: Runtime> { /* private fields */ }

Fuse element wise operations into a single kernel.

Implementations

impl<R: Runtime> ElemwiseOptimization<R>

pub fn new( trace: FuseTrace, client: ComputeClient<R::Server, R::Channel>, device: R::Device, len: usize, ) -> Self

Constructs a new ElemwiseOptimization.

impl<R: Runtime> ElemwiseOptimization<R>

pub fn execute<BT: CubeElement>( &mut self, context: &mut Context<'_, CubeFusionHandle<R>>, )

Execute the optimization.

pub fn num_ops_fused(&self) -> usize

Number of element wise operations fused.

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

Create an optimization from its state.

pub fn to_state(&self) -> ElemwiseOptimizationState

Convert the optimization to its state.