Struct HomogeneousConfig

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

Implementations

impl<M: GlobalConfig> HomogeneousConfig<M>

pub fn new( matmul: M, kernel_size: (u32, u32), stride: (u32, u32), dilation: (u32, u32), padding: (i32, i32), ) -> Self

pub fn to_matmul_config(self) -> M

Trait Implementations

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

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<M: GlobalConfig> ConvGemmConfig for HomogeneousConfig<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

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

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

Formats the value using the given formatter.

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

type Target = M

The resulting type after dereferencing.

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

Dereferences the value.

impl<M: GlobalConfig> GlobalConfig for HomogeneousConfig<M>

type SmmConfig = <M as GlobalConfig>::SmmConfig

Underlying Stage matmul config

fn to_smm_config(&self) -> Self::SmmConfig

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 tiling_dimensions<I: Into<Ident>>(&self, ident: I) -> TilingDimensions

Returns the [StageTiling] for the given ident

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

Returns the MatrixLayout for the given ident

fn num_planes(&self) -> u32

Returns the number of planes in the cube

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

impl<M: Hash + GlobalConfig> Hash for HomogeneousConfig<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 HomogeneousConfig<M>

fn eq(&self, other: &HomogeneousConfig<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 HomogeneousConfig<M>

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

impl<M: GlobalConfig> MatmulConfig for HomogeneousConfig<M>

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