Struct ConvolutionConfig

pub struct ConvolutionConfig<M: GlobalConfig> { /* private fields */ }

Implementations

impl<M: GlobalConfig> ConvolutionConfig<M>

pub fn new( matmul: M, kernel_size: &[u32], stride: &[u32], dilation: &[u32], padding: &[i32], dim: Dimensionality, num_stages: u32, ) -> Result<Self, MatmulSetupError>

pub fn to_matmul_config(self) -> M

Trait Implementations

impl<M: Clone + GlobalConfig> Clone for ConvolutionConfig<M>

fn clone(&self) -> ConvolutionConfig<M>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<M: GlobalConfig> ConvGemmConfig for ConvolutionConfig<M>

fn kernel_size(&self, dim: u32) -> u32

The size of the convolution kernel at dim

fn dilation(&self, dim: u32) -> u32

The dilation of the kernel at dim

fn stride(&self, dim: u32) -> u32

The stride of the kernel at dim

fn padding(&self, dim: u32) -> i32

The padding of the kernel at dim

fn dimensionality(&self) -> Dimensionality

The dimensionality of the kernel

fn line_sizes(&self) -> MatmulLineSizes

impl<M: Debug + GlobalConfig> Debug for ConvolutionConfig<M>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<M: GlobalConfig> Deref for ConvolutionConfig<M>

type Target = M

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl<M: GlobalConfig> GlobalConfig for ConvolutionConfig<M>

type StageConfig = <M as GlobalConfig>::StageConfig

Underlying Stage matmul config

fn stage_config(&self) -> Self::StageConfig

Convert itself to the underlying stage matmul config

fn global_line_size<I: Into<Ident>>(&self, ident: I) -> u32

Returns the line size for the global memory corresponding to the given ident

fn matrix_layout<I: Into<Ident>>(&self, ident: I) -> MatrixLayout

Returns the MatrixLayout for the given ident

fn num_loading_planes<I: Into<Ident>>(&self, ident: I) -> u32

Returns the number of planes participating in loading ident

fn plane_dim(&self) -> u32

Returns the size of the plane dimension

fn check_row_bounds<I: Into<Ident>>(&self, ident: I) -> bool

Whether to check if accessing a row would exceed bounds.

fn check_col_bounds<I: Into<Ident>>(&self, ident: I) -> bool

Whether to check if accessing a col would exceed bounds.

fn check_k_bounds(&self) -> bool

Whether to check if accessing a col for lhs or row for rhs would exceed bounds.

fn precompute_job(&self) -> bool

Whether to put common computations for loading tasks once before loop

fn num_stages(&self, _ident: InputIdent) -> u32

The number of stages in stage memory

fn loader_mode(&self) -> LoaderMode

Whether to check loader is balanced in comptime or runtime.

fn tiling_scheme(&self) -> TilingScheme

Returns the TilingScheme

fn event_loading_mode(&self, ident: InputIdent) -> EventLoadingMode

Whether event loading is constrained to be ordered

fn plane_role_config(&self) -> PlaneRoleConfig

Indicates the specialization roles for the planes

fn specialized_loading_sides(&self) -> SpecializedLoadingSides

Indicates plane roles are associated to loading which tensor input

fn cube_dim(&self) -> CubeDim

The CubeDim arising from the TilingScheme

fn role_rule_config(&self) -> RoleRuleConfig

How to identify the role of the plane depending on its index

fn quantized(&self) -> bool

Whether the matmul is quantized

impl<M: Hash + GlobalConfig> Hash for ConvolutionConfig<M>

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl<M: PartialEq + GlobalConfig> PartialEq for ConvolutionConfig<M>

fn eq(&self, other: &ConvolutionConfig<M>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<M: Copy + GlobalConfig> Copy for ConvolutionConfig<M>

impl<M: Eq + GlobalConfig> Eq for ConvolutionConfig<M>

impl<M: GlobalConfig> StructuralPartialEq for ConvolutionConfig<M>