Struct TiledLayout 

pub struct TiledLayout { /* private fields */ }

Tiling as a composable layout: splits each tiled axis into levels + 1 factors — one grid plus levels nested tile sizes — via a mixed-radix decomposition, expanding logical rank R to physical rank R + levels * n. The factors are read from the physical shape, laid out [pre, grid…, level1…, …, levelL…, post] (coarsest grid first, finest tile last). levels == 1 is a single split c -> (c / T, c % T).

Implementations

impl TiledLayout

pub fn new( physical_shape: CoordsDyn, start_axis: usize, num_tiled: usize, levels: usize, ) -> TiledLayout

pub fn __expand_new( scope: &Scope, physical_shape: <CoordsDyn as CubeType>::ExpandType, start_axis: usize, num_tiled: usize, levels: usize, ) -> <TiledLayout as CubeType>::ExpandType

Trait Implementations

impl Clone for TiledLayout

fn clone(&self) -> TiledLayout

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for TiledLayout

type ExpandType = TiledLayoutExpand

impl Layout for TiledLayout

fn to_source_pos(&self, pos: Self::Coordinates) -> Self::SourceCoordinates

Logical coords to physical tile coords, laid out [pre, grid…, level1…, …, levelL…, post]. Each tiled axis is decomposed mixed-radix: the finest level (last block) is the least significant digit.

fn shape(&self) -> Self::Coordinates

Logical shape: each tiled axis collapses its levels + 1 factors back to their product.

type Coordinates = Sequence<u32>

The coordinate type used by the conceptual tensor represented by this layout, i.e. (u32, u32, u32) for a fixed-rank 3D tensor. This does not have to match the rank of the underlying storage (if applicable). It’s only how the tensor is interpreted (viewed) by the code.

type SourceCoordinates = Sequence<u32>

The coordinate type used by the inner storage wrapped in this layout, i.e. u32 for Array, or (u32, u32) for a 2D view.

fn to_source_pos_checked( &self, pos: Self::Coordinates, ) -> (Self::SourceCoordinates, bool)

Transform a set of n-dimensional coordinates to an offset into the underlying storage, and return whether the position is in bounds of this layout. See also Layout::to_source_pos

fn is_in_bounds(&self, pos: Self::Coordinates) -> bool

fn __expand_to_source_pos( scope: &Scope, this: &<Self as CubeType>::ExpandType, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> <Self::SourceCoordinates as CubeType>::ExpandType

fn __expand_to_source_pos_checked( scope: &Scope, this: &<Self as CubeType>::ExpandType, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> (<Self::SourceCoordinates as CubeType>::ExpandType, <bool as CubeType>::ExpandType)

fn __expand_shape( scope: &Scope, this: &<Self as CubeType>::ExpandType, ) -> <Self::Coordinates as CubeType>::ExpandType

fn __expand_is_in_bounds( scope: &Scope, this: &<Self as CubeType>::ExpandType, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType