Struct SimpleConvAlgorithm

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

Cmma convolution

Trait Implementations

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

type TileMatmul = TMM

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

type GlobalConvolution = SimpleConvolutionFamily<<SimpleConvAlgorithm<TMM> as Algorithm>::StageMatmul>

type Args = TensorArgs

fn cube_dim(selection: &MatmulSelection) -> CubeDim

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 make_config( input: <Self::GlobalConvolution as ConvolutionConfigFactory>::Input, problem: &ConvolutionProblem, cube_dim: &CubeDim, cube_count: &CubeCount, ) -> Result<<Self::GlobalConvolution as ConvolutionConfigFactory>::Config, InvalidConfigError>

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

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