Trait AutotuneOperationSet

pub trait AutotuneOperationSet<K: Send + Sync + 'static, Output = ()>: Send + Sync {
    // 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 methods
    fn compute_checksum(&self) -> String { ... }
    fn should_run(&self, key: &K, index: usize) -> bool { ... }
}

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 Nb: The 0 index is used a “good default”.

Provided Methods

fn compute_checksum(&self) -> String

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

fn should_run(&self, key: &K, index: usize) -> bool

Enable or disable certain indices from being benchmarked based on the key

Implementors