Struct TmaTilingOrder

pub struct TmaTilingOrder;

A special tiling order where:

• If the matrix data layout is row-major, the tiling order is col-major
• If the matrix data layout is col-major, the tiling order is row-major

Trait Implementations

impl Clone for TmaTilingOrder

fn clone(&self) -> TmaTilingOrder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for TmaTilingOrder

type ExpandType = TmaTilingOrderExpand

fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl TilingOrder for TmaTilingOrder

fn to_row_col<C: StageConfig>( nth: u32, tile_count_rows: u32, tile_count_cols: u32, ident: Ident, config: C, ) -> (u32, u32)

Returns the coordinates (row, col) of the tile

fn to_nth_tile<C: StageConfig>( row: u32, col: u32, tile_count_rows: u32, tile_count_cols: u32, ident: Ident, config: C, ) -> u32

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

fn to_enum() -> TilingOrderEnum

Return the trait value as enum

fn __expand_to_row_col<C: StageConfig>( scope: &mut Scope, nth: <u32 as CubeType>::ExpandType, tile_count_rows: u32, tile_count_cols: u32, ident: Ident, config: C, ) -> <(u32, u32) as CubeType>::ExpandType

fn __expand_to_nth_tile<C: StageConfig>( scope: &mut Scope, row: <u32 as CubeType>::ExpandType, col: <u32 as CubeType>::ExpandType, tile_count_rows: u32, tile_count_cols: u32, ident: Ident, config: C, ) -> <u32 as CubeType>::ExpandType

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

impl Copy for TmaTilingOrder