Trait ConvolutionFamily

pub trait ConvolutionFamily: ConvolutionConfigFactory<Config: ConvGemmConfig> + ConvolutionLaunch {
    type Convolution<MP: MatmulPrecision>: Convolution<MP, Config = Self::Config>;

    // Required method
    fn filter_line_sizes(
        available_line_sizes: AvailableLineSizes,
    ) -> AvailableLineSizes;
}

Required Associated Types

type Convolution<MP: MatmulPrecision>: Convolution<MP, Config = Self::Config>

Required Methods

fn filter_line_sizes( available_line_sizes: AvailableLineSizes, ) -> AvailableLineSizes

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<SMM> ConvolutionFamily for MultiStageTmaConvolutionFamily<SMM>

where SMM: StageMatmulFamily<LhsReader = FullReaderFamily, RhsReader = FullReaderFamily>,

type Convolution<MP: MatmulPrecision> = MultiStageTmaConvolution<MP, <SMM as StageMatmulFamily>::Matmul<MP, ContiguousTilingLayout<ColMajorTilingOrder>, ContiguousTilingLayout<RowMajorTilingOrder>>>

impl<SMM> ConvolutionFamily for SimpleConvolutionFamily<SMM>

where SMM: StageMatmulFamily<LhsReader = FullReaderFamily, RhsReader = FullReaderFamily>,

type Convolution<MP: MatmulPrecision> = SimpleConvolution<MP, <SMM as StageMatmulFamily>::Matmul<MP, ContiguousTilingLayout<RowMajorTilingOrder>, ContiguousTilingLayout<RowMajorTilingOrder>>>

impl<SMM> ConvolutionFamily for SimpleTmaConvolutionFamily<SMM>

where SMM: StageMatmulFamily<LhsReader = FullReaderFamily, RhsReader = FullReaderFamily>,

type Convolution<MP: MatmulPrecision> = SimpleTmaConvolution<MP, <SMM as StageMatmulFamily>::Matmul<MP, ContiguousTilingLayout<ColMajorTilingOrder>, ContiguousTilingLayout<RowMajorTilingOrder>>>