Enum burn_wgpu::compute::WgpuAutotuneKey

pub enum WgpuAutotuneKey {
    Matmul(MatmulAutotuneKey),
    SumDim(ReduceAutotuneKey),
    MeanDim(ReduceAutotuneKey),
}

Key for all autotune-enabled operations

Variants

Matmul(MatmulAutotuneKey)

Key for matmul operation

SumDim(ReduceAutotuneKey)

Key for sum_dim operations

MeanDim(ReduceAutotuneKey)

Key for mean_dim operations

Trait Implementations

impl<E: WgpuElement + Element, const D: usize> AutotuneOperationSet<WgpuAutotuneKey> for MatmulAutotuneOperationSet<E, D>

fn key(&self) -> WgpuAutotuneKey

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

impl<E: WgpuElement + Element, const D: usize> AutotuneOperationSet<WgpuAutotuneKey> for MeanDimAutotuneOperationSet<E, D>

fn key(&self) -> WgpuAutotuneKey

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

impl<E: WgpuElement + Element, const D: usize> AutotuneOperationSet<WgpuAutotuneKey> for SumDimAutotuneOperationSet<E, D>

fn key(&self) -> WgpuAutotuneKey

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

impl Clone for WgpuAutotuneKey

fn clone(&self) -> WgpuAutotuneKey

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for WgpuAutotuneKey

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

Formats the value using the given formatter.

impl Display for WgpuAutotuneKey

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

Formats the value using the given formatter.

impl Hash for WgpuAutotuneKey

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 WgpuAutotuneKey

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

This method tests for self and other values to be equal, and is used by ==.

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

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

impl AutotuneKey for WgpuAutotuneKey

impl Eq for WgpuAutotuneKey

impl StructuralEq for WgpuAutotuneKey

impl StructuralPartialEq for WgpuAutotuneKey