ConvGemmConfig

cubecl_convolution::components

Trait ConvGemmConfig 

Source 
pub trait ConvGemmConfig:
    Copy
    + Clone
    + Eq
    + PartialEq
    + Hash
    + Debug
    + Send
    + Sync
    + 'static {
    type GlobalMatmulConfig: GlobalConfig;

    // Required methods
    fn matmul_config(&self) -> Self::GlobalMatmulConfig;
    fn convolution_params(&self) -> ConvolutionParams;
    fn line_sizes(&self) -> MatmulLineSizes;
    fn check_spatial_bounds(&self) -> bool;
    fn cube_dim(&self) -> CubeDim;
    fn lhs_global_memory_config(&self) -> GlobalMemoryConfig;
    fn rhs_global_memory_config(&self) -> GlobalMemoryConfig;
    fn out_global_memory_config(&self) -> GlobalMemoryConfig;
}
Expand description

Convolution specific config, extends regular matmul Config

Required Associated Types§

Source

type GlobalMatmulConfig: GlobalConfig

Required Methods§

Source

fn matmul_config(&self) -> Self::GlobalMatmulConfig

Source

fn convolution_params(&self) -> ConvolutionParams

The size of the convolution kernel at dim

Source

fn line_sizes(&self) -> MatmulLineSizes

Source

fn check_spatial_bounds(&self) -> bool

Source

fn cube_dim(&self) -> CubeDim

Source

fn lhs_global_memory_config(&self) -> GlobalMemoryConfig

Source

fn rhs_global_memory_config(&self) -> GlobalMemoryConfig

Source

fn out_global_memory_config(&self) -> GlobalMemoryConfig

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§