Struct CommonGlobalConfig

pub struct CommonGlobalConfig<S: StageConfig> {
    pub smm_config: S,
    pub check_m_bounds: bool,
    pub check_n_bounds: bool,
    pub check_k_bounds: bool,
    pub lhs_layout: MatrixLayout,
    pub rhs_layout: MatrixLayout,
    pub lhs_line_size: u32,
    pub rhs_line_size: u32,
    pub out_line_size: u32,
    pub num_planes: u32,
}

Configuration for the pipelined global matmul

Fields

smm_config: S

check_m_bounds: bool

check_n_bounds: bool

check_k_bounds: bool

lhs_layout: MatrixLayout

rhs_layout: MatrixLayout

lhs_line_size: u32

rhs_line_size: u32

out_line_size: u32

num_planes: u32

Implementations

impl<S: StageConfig> CommonGlobalConfig<S>

pub fn new( smm_config: S, check_m_bounds: bool, check_n_bounds: bool, check_k_bounds: bool, lhs_layout: MatrixLayout, rhs_layout: MatrixLayout, lhs_line_size: u32, rhs_line_size: u32, out_line_size: u32, num_planes: u32, ) -> Self

Trait Implementations

impl<S: Clone + StageConfig> Clone for CommonGlobalConfig<S>

fn clone(&self) -> CommonGlobalConfig<S>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<S: Debug + StageConfig> Debug for CommonGlobalConfig<S>

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

Formats the value using the given formatter.

impl<S: StageConfig> GlobalConfig for CommonGlobalConfig<S>

type SmmConfig = S

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<S: Hash + StageConfig> Hash for CommonGlobalConfig<S>

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<S: PartialEq + StageConfig> PartialEq for CommonGlobalConfig<S>

fn eq(&self, other: &CommonGlobalConfig<S>) -> 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<S: Copy + StageConfig> Copy for CommonGlobalConfig<S>

impl<S: Eq + StageConfig> Eq for CommonGlobalConfig<S>

impl<S: StageConfig> MatmulConfig for CommonGlobalConfig<S>

impl<S: StageConfig> StructuralPartialEq for CommonGlobalConfig<S>