Enum JitAutotuneKey

pub enum JitAutotuneKey {
    Matmul(MatmulAutotuneKey),
    Reduce(ReduceAutotuneKey),
    Conv2d(Conv2dAutotuneKey),
    ConvTranspose2d(ConvTranspose2dAutotuneKey),
}

Key for all autotune-enabled operations

Variants

Matmul(MatmulAutotuneKey)

Key for matmul operation

Reduce(ReduceAutotuneKey)

Key for reduce dim operations

Conv2d(Conv2dAutotuneKey)

Key for convolution operations

ConvTranspose2d(ConvTranspose2dAutotuneKey)

Key for transpose convolution operations

Trait Implementations

impl<Run: JitRuntime, In: JitElement, Out: JitElement, Rd: Reduce> AutotuneOperationSet<JitAutotuneKey> for ReduceOps<Run, In, Out, Rd>

fn key(&self) -> JitAutotuneKey

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

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

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

fn compute_checksum(&self) -> String

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

impl Clone for JitAutotuneKey

fn clone(&self) -> JitAutotuneKey

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for JitAutotuneKey

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for JitAutotuneKey

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for JitAutotuneKey

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

Formats the value using the given formatter.

impl Hash for JitAutotuneKey

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for JitAutotuneKey

fn eq(&self, other: &JitAutotuneKey) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for JitAutotuneKey

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl AutotuneKey for JitAutotuneKey

impl Eq for JitAutotuneKey

impl StructuralPartialEq for JitAutotuneKey