Struct StridedStageMemory 

pub struct StridedStageMemory<ES: Numeric, NS: Size, T: TilingLayout> {
    pub smem: Shared<[Vector<ES, NS>]>,
    pub swizzle: Swizzle,
    /* private fields */
}

Wrapper over the shared memory used for staging, abstracting its layout

Fields

smem: Shared<[Vector<ES, NS>]>

Underlying shared memory

swizzle: Swizzle

Swizzling of the shared memory, if any

Implementations

impl<ES: Numeric, NS: Size, T: TilingLayout> StridedStageMemory<ES, NS, T>

pub fn new(config: StageMemoryConfig) -> StridedStageMemory<ES, NS, T>

Instantiate a new stage memory for the given identifier

pub fn new_aligned( alignment: usize, config: StageMemoryConfig, ) -> StridedStageMemory<ES, NS, T>

Instantiate a new stage memory for the given identifier, with shared memory alignment

pub fn with_buffer_index(&self, buffer_idx: u32) -> Self

pub fn with_layout<TNew: TilingLayout>( &self, ) -> StridedStageMemory<ES, NS, TNew>

Return the same stage but with a different tiling layout. Allows comptime switching tiling.

pub fn get_tile(&self, tile: Coords2d) -> StridedTile<ES, NS>

Get the tile at position (row, col)

pub fn as_slice<N: Size>(&self) -> &[Vector<ES, N>]

Return the whole stage as a slice, for reading

pub fn as_slice_mut<N: Size>(&mut self) -> &mut [Vector<ES, N>]

Return the whole stage as a mutable slice, for loading

pub unsafe fn free(&self)

Frees the shared memory for reuse, if possible on the target runtime.

Safety

Must be used in uniform control flow Must not have any dangling references to this shared memory

pub fn __expand_new( scope: &Scope, config: StageMemoryConfig, ) -> <StridedStageMemory<ES, NS, T> as CubeType>::ExpandType

Instantiate a new stage memory for the given identifier

pub fn __expand_new_aligned( scope: &Scope, alignment: usize, config: StageMemoryConfig, ) -> <StridedStageMemory<ES, NS, T> as CubeType>::ExpandType

Instantiate a new stage memory for the given identifier, with shared memory alignment

pub fn __expand_with_buffer_index( scope: &Scope, this: &<Self as CubeType>::ExpandType, buffer_idx: <u32 as CubeType>::ExpandType, ) -> <Self as CubeType>::ExpandType

pub fn __expand_with_layout<TNew: TilingLayout>( scope: &Scope, this: &<Self as CubeType>::ExpandType, ) -> <StridedStageMemory<ES, NS, TNew> as CubeType>::ExpandType

Return the same stage but with a different tiling layout. Allows comptime switching tiling.

pub fn __expand_get_tile( scope: &Scope, this: &<Self as CubeType>::ExpandType, tile: <Coords2d as CubeType>::ExpandType, ) -> <StridedTile<ES, NS> as CubeType>::ExpandType

Get the tile at position (row, col)

pub fn __expand_as_slice<'infer, 'scope, N: Size>( scope: &'scope Scope, this: &'infer <Self as CubeType>::ExpandType, ) -> &'infer <[Vector<ES, N>] as CubeType>::ExpandType

Return the whole stage as a slice, for reading

pub fn __expand_as_slice_mut<'infer, 'scope, N: Size>( scope: &'scope Scope, this: &'infer mut <Self as CubeType>::ExpandType, ) -> &'infer mut <[Vector<ES, N>] as CubeType>::ExpandType

Return the whole stage as a mutable slice, for loading

pub fn __expand_free(scope: &Scope, this: &<Self as CubeType>::ExpandType)

Frees the shared memory for reuse, if possible on the target runtime.

Safety

Must be used in uniform control flow Must not have any dangling references to this shared memory

Trait Implementations

impl<ES: Clone + Numeric, NS: Clone + Size, T: Clone + TilingLayout> Clone for StridedStageMemory<ES, NS, T>

fn clone(&self) -> StridedStageMemory<ES, NS, T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<ES: Numeric, NS: Size, T: TilingLayout> CubeType for StridedStageMemory<ES, NS, T>

type ExpandType = StridedStageMemoryExpand<ES, NS, T>

impl<ES: Numeric, NS: Size, T: TilingLayout> Stage<ES> for StridedStageMemory<ES, NS, T>

fn tile<Sc: TileScope>(this: &Self, tile: Coords2d) -> Tile<ES, Sc>

fn as_stage_tile<Sc: TileScope>(this: &Self) -> Tile<ES, Sc>

Wrap the whole stage as a TileKind::Stage tile. Only meaningful for read-only stages consumed by the partition matmul; other impls return Tile::new_None().

fn __expand_tile<Sc: TileScope>( scope: &Scope, this: &<Self as CubeType>::ExpandType, tile: <Coords2d as CubeType>::ExpandType, ) -> <Tile<ES, Sc> as CubeType>::ExpandType

fn __expand_as_stage_tile<Sc: TileScope>( scope: &Scope, this: &<Self as CubeType>::ExpandType, ) -> <Tile<ES, Sc> as CubeType>::ExpandType