Struct StridedTilingLayout 

pub struct StridedTilingLayout {}

Tiles follow a strided layout that often mirrors global memory layout. Not all tiles are contiguous in shared memory, but mapping is more direct.

Implementations

impl StridedTilingLayout

pub fn nth_slice<ES: Numeric>( stage: &mut StridedStage<ES, Self>, nth: u32, config: StageMemoryConfig, ) -> SliceMut<Line<ES>>

Returns the nth slice of the stage

pub fn __expand_nth_slice<ES: Numeric>( scope: &mut Scope, stage: <StridedStage<ES, Self> as CubeType>::ExpandType, nth: <u32 as CubeType>::ExpandType, config: StageMemoryConfig, ) -> <SliceMut<Line<ES>> as CubeType>::ExpandType

Trait Implementations

impl<IP: MatrixPrecision> AsyncLoadingJob<IP, StridedTilingLayout> for AsynFullMaximizeSliceLengthJob

fn execute_task<CM: CopyMechanism, G: GlobalConfig>( this: &mut Self, task_id: u32, tensor_reader: &GlobalIterator<Line<IP::Global>>, stage: &mut StridedStage<IP::Stage, StridedTilingLayout>, mechanism: &CM, config: G, )

Execute the task_idth loading task

fn task_count(this: &Self) -> u32

Get the number of tasks

fn __expand_execute_task<CM: CopyMechanism, G: GlobalConfig>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: <u32 as CubeType>::ExpandType, tensor_reader: <GlobalIterator<Line<IP::Global>> as CubeType>::ExpandType, stage: <StridedStage<IP::Stage, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl<IP: MatrixPrecision> AsyncLoadingJob<IP, StridedTilingLayout> for AsyncFullCooperativeJob

fn execute_task<CM: CopyMechanism, G: GlobalConfig>( this: &mut Self, task_id: u32, global_iter: &GlobalIterator<Line<IP::Global>>, stage: &mut StridedStage<IP::Stage, StridedTilingLayout>, mechanism: &CM, config: G, )

Execute the task_idth loading task

fn task_count(this: &Self) -> u32

Get the number of tasks

fn __expand_execute_task<CM: CopyMechanism, G: GlobalConfig>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: <u32 as CubeType>::ExpandType, global_iter: <GlobalIterator<Line<IP::Global>> as CubeType>::ExpandType, stage: <StridedStage<IP::Stage, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl<IP: MatrixPrecision> AsyncLoadingJob<IP, StridedTilingLayout> for AsyncFullMaximizeUnitCountJob

fn execute_task<CM: CopyMechanism, G: GlobalConfig>( this: &mut Self, _task_id: u32, global_iter: &GlobalIterator<Line<IP::Global>>, stage: &mut StridedStage<IP::Stage, StridedTilingLayout>, mechanism: &CM, config: G, )

Execute the task_idth loading task

fn task_count(_this: &Self) -> u32

Get the number of tasks

fn __expand_execute_task<CM: CopyMechanism, G: GlobalConfig>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, _task_id: <u32 as CubeType>::ExpandType, global_iter: <GlobalIterator<Line<IP::Global>> as CubeType>::ExpandType, stage: <StridedStage<IP::Stage, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( scope: &mut Scope, _this: <Self as CubeType>::ExpandType, ) -> u32

impl<IP: MatrixPrecision> AsyncLoadingJob<IP, StridedTilingLayout> for AsyncPartialMaximizeSliceLengthJob

fn execute_task<CM: CopyMechanism, G: GlobalConfig>( this: &mut Self, task_id: u32, global_iter: &GlobalIterator<Line<IP::Global>>, stage: &mut StridedStage<IP::Stage, StridedTilingLayout>, mechanism: &CM, config: G, )

Execute the task_idth loading task

fn task_count(this: &Self) -> u32

Get the number of tasks

fn __expand_execute_task<CM: CopyMechanism, G: GlobalConfig>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: <u32 as CubeType>::ExpandType, global_iter: <GlobalIterator<Line<IP::Global>> as CubeType>::ExpandType, stage: <StridedStage<IP::Stage, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl Clone for StridedTilingLayout

fn clone(&self) -> StridedTilingLayout

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<IP: MatrixPrecision> LoadingJob<IP, StridedTilingLayout> for SyncFullStridedJob

fn execute_task<G: GlobalConfig>( this: &mut Self, task_id: u32, global_iter: &GlobalIterator<Line<IP::Global>>, stage: &mut StridedStage<IP::Stage, StridedTilingLayout>, config: G, )

Execute the task_idth loading task

fn task_count(this: &Self) -> u32

Get the number of tasks

fn __expand_execute_task<G: GlobalConfig>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: u32, global_iter: <GlobalIterator<Line<IP::Global>> as CubeType>::ExpandType, stage: <StridedStage<IP::Stage, StridedTilingLayout> as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl TilingLayout for StridedTilingLayout

fn get_tile<ES: Numeric>( stage: &StridedStage<ES, Self>, tile: Coords2d, _buffer_index: u32, _ident: StageIdent, config: StageMemoryConfig, ) -> StridedTile<ES>

Returns the tile at shared memory coordinates

fn __expand_get_tile<ES: Numeric>( scope: &mut Scope, stage: <StridedStage<ES, Self> as CubeType>::ExpandType, tile: <Coords2d as CubeType>::ExpandType, _buffer_index: <u32 as CubeType>::ExpandType, _ident: StageIdent, config: StageMemoryConfig, ) -> <StridedTile<ES> as CubeType>::ExpandType

impl TilingValidation for StridedTilingLayout

fn check(config: GlobalMemoryConfig) -> Result<(), InvalidConfigError>

impl Copy for StridedTilingLayout