Struct InterleavedTile 

pub struct InterleavedTile<N: Numeric> {
    pub data: Array<N>,
    pub matrix_layout: MatrixLayout,
    pub tile_size: TileSize,
    pub plane_dim: u32,
}

Interleaved-on-k tile. Holds just the minimal comptime data the body uses (tile_size and plane_dim); the matmul-level configuration (and any metadata not consumed by the tile body, like swizzle modes) lives in cubek-matmul as InterleavedMatmul.

Fields

data: Array<N>

matrix_layout: MatrixLayout

tile_size: TileSize

plane_dim: u32

Implementations

impl<A: Numeric> InterleavedTile<A>

pub fn mma<L: Numeric, R: Numeric>( &mut self, lhs: &InterleavedTile<L>, rhs: &InterleavedTile<R>, )

Executes lhs · rhs, accumulating into self via the plane- interleaved-on-k matmul.

pub fn __expand_mma<L: Numeric, R: Numeric>( scope: &Scope, this: &mut <Self as CubeType>::ExpandType, lhs: &<InterleavedTile<L> as CubeType>::ExpandType, rhs: &<InterleavedTile<R> as CubeType>::ExpandType, )

Executes lhs · rhs, accumulating into self via the plane- interleaved-on-k matmul.

impl<N: Numeric> InterleavedTile<N>

pub fn copy_from<SE: Numeric, SS: Size, Sc: TileScope>( &mut self, source: &Tile<SE, Sc>, ident: StageIdent, )

Copies into the interleaved tile from source. Supported sources: Shared and None (zero-init).

pub fn init_zero(&mut self)

pub fn __expand_copy_from<SE: Numeric, SS: Size, Sc: TileScope>( scope: &Scope, this: &mut <Self as CubeType>::ExpandType, source: &<Tile<SE, Sc> as CubeType>::ExpandType, ident: StageIdent, )

Copies into the interleaved tile from source. Supported sources: Shared and None (zero-init).

pub fn __expand_init_zero( scope: &Scope, this: &mut <Self as CubeType>::ExpandType, )

Trait Implementations

impl<N: Numeric> CubeType for InterleavedTile<N>

type ExpandType = InterleavedTileExpand<N>