Struct SimpleAsyncTmaConv 

pub struct SimpleAsyncTmaConv<TMM: TileMatmulFamily> { /* private fields */ }

Trait Implementations

impl<TMM: TileMatmulFamily<LhsTile = Strided, RhsTile = Strided, AccTile = Option<Strided>, OutTile = Strided>> Algorithm for SimpleAsyncTmaConv<TMM>

type Routine = SimpleAlgorithm<TMM, AsyncFullTmaLoading, AsyncFullTmaLoading, SyncBiasLoading>

type Args = TensorMapArgs<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

const IS_SPECIALIZED: bool = false

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