Struct SpecializedConv 

pub struct SpecializedConv<L: AsyncPartialLoadingStrategy<RuntimeArgs>> { /* private fields */ }

Cmma convolution

Trait Implementations

impl<L: AsyncPartialLoadingStrategy<RuntimeArgs, TileKind = Strided>> Algorithm for SpecializedConv<L>

const IS_SPECIALIZED: bool = true

Whether to select specialized load flow in tests. Should replace with something cleaner eventually, but this is nice and simple.

type Routine = SpecializedAlgorithm<L, SyncBiasLoading>

type Args = TensorArgs<RuntimeArgs>

fn correct_layout<R: Runtime>( client: &ComputeClient<R>, handle: TensorBinding<R>, dtype: StorageType, _operation: ConvolutionOperation, ) -> Result<TensorBinding<R>, LaunchError>

fn filter_vector_sizes( vector_sizes: AvailableVectorSizes, ) -> AvailableVectorSizes