Enum CubeOptimization 

pub enum CubeOptimization<R: Runtime> {
    ElementWise(ElemwiseOptimization<R>),
    Matmul(MatmulOptimization<R>),
    Reduce(ReduceOptimization<R>),
    ReduceBroadcasted(ReduceBroadcastedOptimization<R>),
}

Fusion optimization type for cubecl.

More optimization variants should be added here.

Variants

ElementWise(ElemwiseOptimization<R>)

Matmul(MatmulOptimization<R>)

Reduce(ReduceOptimization<R>)

ReduceBroadcasted(ReduceBroadcastedOptimization<R>)

Implementations

impl<R: Runtime> CubeOptimization<R>

pub fn to_opt_state(&self) -> CubeOptimizationState

Serializes the current optimization to its state.

Trait Implementations

impl<R: Runtime> Debug for CubeOptimization<R>

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

Formats the value using the given formatter.

impl<R: Runtime> NumOperations for CubeOptimization<R>

fn len(&self) -> usize

The number of registered operations.

fn name(&self) -> &'static str

The name of the optimization.

fn is_empty(&self) -> bool

If the current optimization is empty.

impl<R: Runtime> OperationFuser<CubeOptimization<R>> for ElementWiseFuser<R>

fn fuse(&mut self, operation: &OperationIr)

Register a new tensor operation.

fn finish(&mut self) -> CubeOptimization<R>

Finish the optimization and create a fusion operation.

fn reset(&mut self)

Reset the state.

fn status(&self) -> FuserStatus

Return the builder status.

fn properties(&self) -> FuserProperties

Return the builder properties.

fn len(&self) -> usize

The number of operation fused.

fn clone_dyn(&self) -> Box<dyn OperationFuser<CubeOptimization<R>>>

Clone the optimization builder.

fn is_empty(&self) -> bool

If no operations are fused.

impl<R: Runtime> OperationFuser<CubeOptimization<R>> for MatmulFuser<R>

fn fuse(&mut self, operation: &OperationIr)

Register a new tensor operation.

fn finish(&mut self) -> CubeOptimization<R>

Finish the optimization and create a fusion operation.

fn reset(&mut self)

Reset the state.

fn status(&self) -> FuserStatus

Return the builder status.

fn properties(&self) -> FuserProperties

Return the builder properties.

fn len(&self) -> usize

The number of operation fused.

fn clone_dyn(&self) -> Box<dyn OperationFuser<CubeOptimization<R>>>

Clone the optimization builder.

fn is_empty(&self) -> bool

If no operations are fused.

impl<R: Runtime> OperationFuser<CubeOptimization<R>> for ReduceFuser<R>

fn fuse(&mut self, operation: &OperationIr)

Register a new tensor operation.

fn finish(&mut self) -> CubeOptimization<R>

Finish the optimization and create a fusion operation.

fn reset(&mut self)

Reset the state.

fn status(&self) -> FuserStatus

Return the builder status.

fn properties(&self) -> FuserProperties

Return the builder properties.

fn len(&self) -> usize

The number of operation fused.

fn clone_dyn(&self) -> Box<dyn OperationFuser<CubeOptimization<R>>>

Clone the optimization builder.

fn is_empty(&self) -> bool

If no operations are fused.

impl<R: Runtime> OperationFuser<CubeOptimization<R>> for ReduceBroadcastedFuser<R>

fn fuse(&mut self, operation: &OperationIr)

Register a new tensor operation.

fn finish(&mut self) -> CubeOptimization<R>

Finish the optimization and create a fusion operation.

fn reset(&mut self)

Reset the state.

fn status(&self) -> FuserStatus

Return the builder status.

fn properties(&self) -> FuserProperties

Return the builder properties.

fn len(&self) -> usize

The number of operation fused.

fn clone_dyn(&self) -> Box<dyn OperationFuser<CubeOptimization<R>>>

Clone the optimization builder.

fn is_empty(&self) -> bool

If no operations are fused.