Trait cubecl_runtime::tune::AutotuneOperationSet

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

    // 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>>

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>

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