Trait cubecl_runtime::tune::AutotuneOperationSet

pub trait AutotuneOperationSet<K, Output = ()>: Send {
    // Required methods
    fn key(&self) -> K;
    fn autotunables(&self) -> Vec<Box<dyn AutotuneOperation<Output>>>;
    fn fastest(
        self: Box<Self>,
        fastest_index: usize,
    ) -> Box<dyn AutotuneOperation<Output>>;

    // Provided method
    fn compute_checksum(&self) -> String { ... }
}

Groups operations of the same type for autotune

Required Methods

fn key(&self) -> K

The key used in the tune cache

fn autotunables(&self) -> Vec<Box<dyn AutotuneOperation<Output>>>

All candidate operations for autotuning this operation type Operations can run on toy tensors of relevant size

fn fastest( self: Box<Self>, fastest_index: usize, ) -> Box<dyn AutotuneOperation<Output>>

Returns the operation for the given index, matching the order returned by autotunables. Operation obtained here runs on original tensors

Provided Methods

fn compute_checksum(&self) -> String

Compute a checksum that can invalidate outdated cached auto-tune results.

Implementors