Struct CubeOptimization 

pub struct CubeOptimization { /* private fields */ }

Available on crate features fusion only.

A fusion optimization ready to run on an execution stream: what fusers finish and the fusion runtime executes. Wraps any FusedOperation.

Implementations

impl CubeOptimization

pub fn new(optimization: impl FusedOperation) -> CubeOptimization

Wrap the fused operation.

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

The optimization’s name.

pub fn num_ops_fused(&self) -> usize

The number of operations fused.

pub fn run( &mut self, context: &mut Context<CubeFusionHandle>, fallback: &dyn Fn(usize) -> Box<dyn FallbackOperation>, )

Run the fused operation. See FusedOperation::run.

pub fn to_state(&self) -> CubeOptimizationState

Serialize the optimization: its name plus its state encoded as bytes.

Trait Implementations

impl Debug for CubeOptimization

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

Formats the value using the given formatter.

impl NumOperations for CubeOptimization

fn len(&self) -> usize

The number of registered operations.

fn max_relative_shape_id(&self) -> Option<usize>

The highest relative shape id this optimization names, if it names any. None means the optimization names no relative shape, so it fits any stream.

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

The name of the optimization.

fn is_empty(&self) -> bool

If the current optimization is empty.

impl Optimization<FusionCubeRuntime> for CubeOptimization

fn execute( &mut self, context: &mut Context<<FusionCubeRuntime as FusionRuntime>::FusionHandle>, execution: &OrderedExecution<FusionCubeRuntime>, )

Execute the optimization.

fn to_state(&self) -> CubeOptimizationState

Returns the state that can be serialized.

fn from_state(device: &CubeDevice, state: CubeOptimizationState) -> Self

Create the optimization from the state.