Enum CubeOptimization

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

Fusion optimization type for cubecl.

More optimization variants should be added here.

Variants

ElementWise(ElemwiseOptimization<R>)

Element wise optimization.

Matmul(MatmulOptimization<R>)

Matrix multiplication optimization.

Reduce(ReduceOptimization<R>)

Implementations

impl<R: Runtime> CubeOptimization<R>

pub fn to_opt_state(&self) -> CubeOptimizationState

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 is_empty(&self) -> bool

If the current optimization is empty.

impl<R: Runtime> OptimizationBuilder<CubeOptimization<R>> for ElementWiseBuilder<R>

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

Register a new tensor operation.

fn build(&self) -> CubeOptimization<R>

Finish the optimization and create a fusion operation.

fn reset(&mut self)

Reset the state.

fn status(&self) -> OptimizationStatus

Return the builder status.

fn properties(&self) -> OptimizationProperties

Return the builder properties.

fn len(&self) -> usize

The number of operation fused.

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

Clone the optimization builder.

fn is_empty(&self) -> bool

If no operations are fused.

impl<R: Runtime> OptimizationBuilder<CubeOptimization<R>> for MatmulBuilder<R>

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

Register a new tensor operation.

fn build(&self) -> CubeOptimization<R>

Finish the optimization and create a fusion operation.

fn reset(&mut self)

Reset the state.

fn status(&self) -> OptimizationStatus

Return the builder status.

fn properties(&self) -> OptimizationProperties

Return the builder properties.

fn len(&self) -> usize

The number of operation fused.

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

Clone the optimization builder.

fn is_empty(&self) -> bool

If no operations are fused.

impl<R: Runtime> OptimizationBuilder<CubeOptimization<R>> for ReduceBuilder<R>

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

Register a new tensor operation.

fn build(&self) -> CubeOptimization<R>

Finish the optimization and create a fusion operation.

fn reset(&mut self)

Reset the state.

fn status(&self) -> OptimizationStatus

Return the builder status.

fn properties(&self) -> OptimizationProperties

Return the builder properties.

fn len(&self) -> usize

The number of operation fused.

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

Clone the optimization builder.

fn is_empty(&self) -> bool

If no operations are fused.