TilingOrder

cubecl_matmul::components::stage

Trait TilingOrder 

Source 
pub trait TilingOrder:
    'static
    + Send
    + Sync
    + Clone
    + Copy {
    // Required methods
    fn to_row_col(
        nth: u32,
        tile_count_rows: u32,
        tile_count_cols: u32,
        config: StageMemoryConfig,
    ) -> Coords2d;
    fn to_nth_tile(
        tile: Coords2d,
        tile_count_rows: u32,
        tile_count_cols: u32,
        config: StageMemoryConfig,
    ) -> u32;
    fn to_enum() -> TilingOrderEnum;
    fn __expand_to_row_col(
        scope: &mut Scope,
        nth: <u32 as CubeType>::ExpandType,
        tile_count_rows: <u32 as CubeType>::ExpandType,
        tile_count_cols: <u32 as CubeType>::ExpandType,
        config: StageMemoryConfig,
    ) -> <Coords2d as CubeType>::ExpandType;
    fn __expand_to_nth_tile(
        scope: &mut Scope,
        tile: <Coords2d as CubeType>::ExpandType,
        tile_count_rows: <u32 as CubeType>::ExpandType,
        tile_count_cols: <u32 as CubeType>::ExpandType,
        config: StageMemoryConfig,
    ) -> <u32 as CubeType>::ExpandType;
    fn __expand_to_enum(scope: &mut Scope) -> TilingOrderEnum;
}
Expand description

Determines the order in which tiles are stored in shared memory, if TilingLayout is contiguous

Required Methods§

Source

fn to_row_col( nth: u32, tile_count_rows: u32, tile_count_cols: u32, config: StageMemoryConfig, ) -> Coords2d

Returns the coordinates (row, col) of the tile

Source

fn to_nth_tile( tile: Coords2d, tile_count_rows: u32, tile_count_cols: u32, config: StageMemoryConfig, ) -> u32

Given the coordinates (row, col) of the tile, returns its index in shared memory

Source

fn to_enum() -> TilingOrderEnum

Return the trait value as enum

Source

fn __expand_to_row_col( scope: &mut Scope, nth: <u32 as CubeType>::ExpandType, tile_count_rows: <u32 as CubeType>::ExpandType, tile_count_cols: <u32 as CubeType>::ExpandType, config: StageMemoryConfig, ) -> <Coords2d as CubeType>::ExpandType

Source

fn __expand_to_nth_tile( scope: &mut Scope, tile: <Coords2d as CubeType>::ExpandType, tile_count_rows: <u32 as CubeType>::ExpandType, tile_count_cols: <u32 as CubeType>::ExpandType, config: StageMemoryConfig, ) -> <u32 as CubeType>::ExpandType

Source

fn __expand_to_enum(scope: &mut Scope) -> TilingOrderEnum

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§