cubecl_convolution::base

Trait ConvolutionConfigFactory

Source 
pub trait ConvolutionConfigFactory:
    Send
    + Sync
    + 'static {
    type Config: ConvGemmConfig;

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

Provides configuration for a matmul kernel at any level

Required Associated Types§

Source

type Config: ConvGemmConfig

Configuration tailored to the matmul implementation

Required Methods§

Source

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

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§