Struct SimpleTmaConvolutionFamily

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

Trait Implementations

impl<SMM> ConvolutionConfigFactory for SimpleTmaConvolutionFamily<SMM>

where SMM: StageMatmulFamily,

type Config = HomogeneousConfig<Config<<SMM as MatmulConfigFactory>::Config>>

Configuration tailored to the matmul implementation

type Input = <SMM as MatmulConfigFactory>::Input

fn check_config(config: &Self::Config) -> Result<(), InvalidConfigError>

Asserts that the configuration for this matmul will lead to a valid computation

fn make_config( input: Self::Input, problem: &ConvolutionProblem, cube_dim: &CubeDim, cube_count: &CubeCount, ) -> Self::Config

fn check_availability<R: Runtime, MP: MatmulPrecision>( client: &ComputeClient<R::Server, R::Channel>, config: &Self::Config, ) -> Result<(), MatmulAvailabilityError>

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

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