Trait Algorithm 

pub trait Algorithm {
    type TileMatmul: TileMatmulFamily;
    type StageMatmul: StageMatmulFamily;
    type GlobalConvolution: GlobalConvolutionFamily;
    type Args: MatmulArgs;

    // Required methods
    fn num_stages() -> NumStages;
    fn into_tensor_handle<R: Runtime, E: Numeric>(
        client: &ComputeClient<R::Server>,
        handle: &TensorHandleRef<'_, R>,
        ident: MatmulIdent,
    ) -> TensorHandle<R, E>;
    fn selection<R: Runtime>(
        client: &ComputeClient<R::Server>,
        problem: &ConvolutionProblem,
        plane_dim: u32,
        matmul_elems: MatmulElems,
    ) -> Result<MatmulSelection, MatmulSetupError>;

    // Provided methods
    fn cube_count(
        selection: &MatmulSelection,
        problem: &ConvolutionProblem,
    ) -> CubeCount { ... }
    fn multi_row_strategy() -> MultiRowStrategy { ... }
    fn loading_precompute_strategy() -> LoadingPrecomputeStrategy { ... }
    fn reader_mode() -> ReaderMode { ... }
    fn load_specialization() -> LoadSpecializationConfig { ... }
    fn partition_buffering_strategy() -> PartitionBuffering { ... }
    fn setup<R: Runtime, MP: MatmulPrecision>(
        client: &ComputeClient<R::Server>,
        problem: &ConvolutionProblem,
        selection: &MatmulSelection,
        line_sizes: &MatmulLineSizes,
    ) -> Result<GlobalConfig<Self::GlobalConvolution>, MatmulSetupError> { ... }
    fn filter_line_sizes(
        available_line_sizes: AvailableLineSizes,
    ) -> AvailableLineSizes { ... }
}

Specifications for a convolution algorithm

Required Associated Types

type TileMatmul: TileMatmulFamily

type StageMatmul: StageMatmulFamily

type GlobalConvolution: GlobalConvolutionFamily

type Args: MatmulArgs

Required Methods

fn num_stages() -> NumStages

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

fn selection<R: Runtime>( client: &ComputeClient<R::Server>, problem: &ConvolutionProblem, plane_dim: u32, matmul_elems: MatmulElems, ) -> Result<MatmulSelection, MatmulSetupError>

Provided Methods

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

fn multi_row_strategy() -> MultiRowStrategy

fn loading_precompute_strategy() -> LoadingPrecomputeStrategy

fn reader_mode() -> ReaderMode

fn load_specialization() -> LoadSpecializationConfig

fn partition_buffering_strategy() -> PartitionBuffering

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

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

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

Implementors

impl<TMM: TileMatmulFamily<LhsTile = Strided, RhsTile = Strided, AccTile = CubeOption<Strided>, OutTile = Strided>> Algorithm for MultiStageTmaConvAlgorithm<TMM>

type TileMatmul = TMM

type StageMatmul = PlaneMatmulFamily<<MultiStageTmaConvAlgorithm<TMM> as Algorithm>::TileMatmul, StridedStageFamily, StridedStageFamily, Option<StridedStageFamily>>

type GlobalConvolution = MultiStageTmaConvolutionFamily<<MultiStageTmaConvAlgorithm<TMM> as Algorithm>::StageMatmul>

type Args = TensorMapArgs

impl<TMM: TileMatmulFamily<LhsTile = Strided, RhsTile = Strided, AccTile = CubeOption<Strided>, OutTile = Strided>> Algorithm for SimpleConvAlgorithm<TMM>

type TileMatmul = TMM

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

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

type Args = TensorArgs

impl<TMM: TileMatmulFamily<LhsTile = Strided, RhsTile = Strided, AccTile = CubeOption<Strided>, OutTile = Strided>> Algorithm for SimpleTmaConvAlgorithm<TMM>

type TileMatmul = TMM

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

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

type Args = TensorMapArgs