Struct SimpleTmaConvAlgorithm

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

Cmma convolution

Trait Implementations

impl<TMM: TileMatmulFamily> Algorithm for SimpleTmaConvAlgorithm<TMM>

type TileMatmul = TMM

type StageMatmul = PlaneMatmulFamily<<SimpleTmaConvAlgorithm<TMM> as Algorithm>::TileMatmul, FullReaderFamily, FullReaderFamily>

type GlobalConvolution = SimpleTmaConvolutionFamily<<SimpleTmaConvAlgorithm<TMM> as Algorithm>::StageMatmul>

type Args = TensorMapArgs

fn cube_count( selection: &MatmulSelection, problem: &ConvolutionProblem, ) -> CubeCount

fn into_tensor_handle<R: Runtime, E: Numeric>( client: &ComputeClient<R::Server, R::Channel>, handle: &TensorHandleRef<'_, R>, ident: InputIdent, ) -> TensorHandle<R, E>

fn num_stages() -> NumStages

fn selection<R: Runtime>( client: &ComputeClient<R::Server, R::Channel>, problem: &ConvolutionProblem, plane_dim: u32, elem_stage: Elem, elem_acc: Elem, ) -> MatmulSelection

fn multi_row_strategy() -> MultiRowStrategy

fn loading_precompute_strategy() -> LoadingPrecomputeStrategy

fn loader_mode() -> LoaderMode

fn load_specialization() -> LoadSpecializationConfig

fn partition_buffering_strategy() -> PartitionBuffering

fn setup<R: Runtime, MP: MatmulPrecision>( client: &ComputeClient<R::Server, R::Channel>, problem: &ConvolutionProblem, selection: &MatmulSelection, line_sizes: &MatmulLineSizes, ) -> Result<<Self::GlobalConvolution as ConvolutionConfigFactory>::Config, MatmulSetupError>

Make a convolution config from a convolution problem, and launch options

fn filter_line_sizes( available_line_sizes: AvailableLineSizes, ) -> AvailableLineSizes