Struct TiledViewLayout 

pub struct TiledViewLayout { /* private fields */ }

Tiling above a strided buffer: TiledLayout’s split, then per-axis strides to a linear offset. The buffer is a plain strided tensor; tiling is explicit (start_axis / num_tiled), with no Tiler metadata on it.

Trait Implementations

impl Clone for TiledViewLayout

fn clone(&self) -> TiledViewLayout

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for TiledViewLayout

type ExpandType = TiledViewLayoutExpand

impl Layout for TiledViewLayout

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

Logical coords to buffer offset: split to physical coords, then strides.

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 = usize

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 shape(&self) -> Self::Coordinates

The shape of the conceptual tensor represented by this layout. Not necessarily the extent of the underlying storage, but only this view of it.

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

impl ViewLayoutLaunchArg for TiledViewLayout

type RuntimeArg<R: Runtime> = TileSpec

The runtime argument for the kernel.

type CompilationArg = TiledViewLayoutCompilationArg

Compilation argument.

fn register<R: Runtime, B: MemoryArg>( spec: Self::RuntimeArg<R>, buffer: &B, _ty: Type, launcher: &mut KernelLauncher<R>, ) -> Self::CompilationArg

fn expand( arg: &Self::CompilationArg, _ty: Type, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( arg: &Self::CompilationArg, ty: Type, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an output variable during compilation that fill the KernelBuilder.