Struct Stage

pub struct Stage<ES: Numeric, T: TilingLayout> { /* private fields */ }

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

Implementations

impl<ES: Numeric, T: TilingLayout> Stage<ES, T>

pub fn new<S: StageConfig>(ident: Ident, config: S) -> Stage<ES, T>

Instantiate a new stage for the given identifier

pub fn new_aligned<S: StageConfig>( ident: Ident, alignment: u32, config: S, ) -> Stage<ES, T>

Instantiate a new stage for the given identifier

pub fn new_with_smem(smem: SharedMemory<Line<ES>>) -> Stage<ES, T>

Instantiate with a custom shared memory

pub fn get_tile<S: StageConfig>( &self, x: u32, y: u32, ident: Ident, config: S, ) -> Tile<ES>

Get the tile at position (x,y) regardless of matrix layout

pub fn as_slice(&self, line_size: u32) -> Slice<Line<ES>>

Return the whole stage as a slice, for reading

pub fn as_slice_mut(&mut self, line_size: u32) -> SliceMut<Line<ES>>

Return the whole stage as a mutable slice, for loading

pub fn clear<S: StageConfig>(&mut self, ident: InputIdent, config: S)

pub fn clear_buffer<S: StageConfig>( &mut self, buffer_id: BufferId, ident: InputIdent, config: S, )

pub fn __expand_new<S: StageConfig>( context: &mut Scope, ident: Ident, config: S, ) -> <Stage<ES, T> as CubeType>::ExpandType

pub fn __expand_new_aligned<S: StageConfig>( context: &mut Scope, ident: Ident, alignment: u32, config: S, ) -> <Stage<ES, T> as CubeType>::ExpandType

pub fn __expand_new_with_smem( context: &mut Scope, smem: <SharedMemory<Line<ES>> as CubeType>::ExpandType, ) -> <Stage<ES, T> as CubeType>::ExpandType

pub fn __expand_get_tile<S: StageConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, x: <u32 as CubeType>::ExpandType, y: <u32 as CubeType>::ExpandType, ident: Ident, config: S, ) -> <Tile<ES> as CubeType>::ExpandType

pub fn __expand_as_slice( context: &mut Scope, this: <Self as CubeType>::ExpandType, line_size: u32, ) -> <Slice<Line<ES>> as CubeType>::ExpandType

pub fn __expand_as_slice_mut( context: &mut Scope, this: <Self as CubeType>::ExpandType, line_size: u32, ) -> <SliceMut<Line<ES>> as CubeType>::ExpandType

pub fn __expand_clear<S: StageConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, ident: InputIdent, config: S, ) -> <() as CubeType>::ExpandType

pub fn __expand_clear_buffer<S: StageConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, buffer_id: BufferId, ident: InputIdent, config: S, ) -> <() as CubeType>::ExpandType

Trait Implementations

impl<ES: Clone + Numeric, T: Clone + TilingLayout> Clone for Stage<ES, T>

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<ES: Numeric, T: TilingLayout> CubeType for Stage<ES, T>

type ExpandType = StageExpand<ES, T>

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

Wrapper around the init method, necessary to type inference.

impl<ES: Copy + Numeric, T: Copy + TilingLayout> Copy for Stage<ES, T>