cubecl_linalg::matmul::components::tile

Trait TileConfig

Source 
pub trait TileConfig: MatmulConfig {
    // Required methods
    fn plane_dim(&self) -> u32;
    fn matrix_layout(&self, ident: Ident) -> MatrixLayout;
    fn stage_line_size(&self, ident: Ident) -> u32;
    fn tile_shape(&self) -> &MatmulSize;
}
Expand description

Configuration for the Tile matmul (TMM) level

Required Methods§

Source

fn plane_dim(&self) -> u32

Returns the size of the plane dimension

Source

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

Returns the MatrixLayout for the given ident

Source

fn stage_line_size(&self, ident: Ident) -> u32

Returns the line size for the given ident

Source

fn tile_shape(&self) -> &MatmulSize

Returns the shape of the tiles in the three axes m, k and n.

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§