Struct SimpleTmaConvolutionFamily

pub struct SimpleTmaConvolutionFamily<SMM: StageMatmulFamily> { /* private fields */ }

Trait Implementations

impl<SMM> ConvolutionConfigFactory for SimpleTmaConvolutionFamily<SMM>

where SMM: StageMatmulFamily,

type Config = ConvolutionConfig<SimpleTmaConfig<<SMM as StageMatmulFamily>::Config>>

Configuration tailored to the matmul implementation

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

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>>>

fn filter_line_sizes( available_line_sizes: AvailableLineSizes, ) -> AvailableLineSizes

impl<SMM: StageMatmulFamily<LhsReader = FullReaderFamily, RhsReader = FullReaderFamily>> ConvolutionLaunch for SimpleTmaConvolutionFamily<SMM>

unsafe fn launch_unchecked<'a, MS: MatmulSpec, R: Runtime>( client: &ComputeClient<<R as Runtime>::Server, <R as Runtime>::Channel>, cube_dim: CubeDim, cube_count: CubeCount, input: InputRuntimeArg<'a, MS, R>, bias: Option<TensorArg<'a, R>>, output: OutputRuntimeArg<'a, MS, R>, problem: &ConvolutionProblem, config: <Self as ConvolutionConfigFactory>::Config, )

Entry point