Struct SyncFullTilewiseLoading

pub struct SyncFullTilewiseLoading<T: TilingOrder> { /* private fields */ }

Each tile is guaranteed to be loaded entirely by the same plane. Each plane can load multiple tiles, provided the number of planes evenly divides the number of tiles. In this case, a plane loads contiguous tiles following the TilingOrder.

If number of planes = number of rows of Lhs and TilingOrder is RowMajor, each plane loads its own row and a sync can be saved. In multi-row, number of planes must divide number of rows, and each plane loads a contiguous chunk of rows (e.g. plane 0 loads rows 0–1, plane 1 loads 2–3, etc.).

Trait Implementations

impl<T: Clone + TilingOrder> Clone for SyncFullTilewiseLoading<T>

fn clone(&self) -> SyncFullTilewiseLoading<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: TilingOrder> CubeType for SyncFullTilewiseLoading<T>

type ExpandType = SyncFullTilewiseLoadingExpand<T>

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

Wrapper around the init method, necessary to type inference.

impl<TO: TilingOrder> LoadMaxRoundPlaneCount for SyncFullTilewiseLoading<TO>

fn max_round_plane_count( tiling_scheme: &TilingScheme, ident: InputIdent, _line_size: u8, _plane_dim: u32, ) -> u32

Returns the largest number of planes that evenly divides the tasks.

impl<T: TilingOrder> LoadingValidation for SyncFullTilewiseLoading<T>

fn check<C: GlobalConfig>( config: &C, ident: Ident, ) -> Result<(), InvalidConfigError>

Verify that configs are valid for a loader, otherwise return an error stating why

impl<TO: TilingOrder> SyncFullLoadingStrategy for SyncFullTilewiseLoading<TO>

type TilingLayout = ContiguousTilingLayout<TO>

The layout describing how data is tiled across the stage.

type Job<MP: MatmulPrecision> = SyncFullTilewiseJob

The LoadingJob for this strategy.

fn new_job<MP: MatmulPrecision, G: GlobalConfig>( input_ident: InputIdent, config: G, ) -> Self::Job<MP>

Returns the job with preliminary calculations done.

fn __expand_new_job<MP: MatmulPrecision, G: GlobalConfig>( scope: &mut Scope, input_ident: InputIdent, config: G, ) -> <Self::Job<MP> as CubeType>::ExpandType

impl<T: Copy + TilingOrder> Copy for SyncFullTilewiseLoading<T>