Struct InterleavedTileExpand 

pub struct InterleavedTileExpand<N: Numeric> {
    pub data: <Array<N> as CubeType>::ExpandType,
    pub matrix_layout: MatrixLayout,
    pub tile_size: TileSize,
    pub plane_dim: u32,
}

Fields

data: <Array<N> as CubeType>::ExpandType

matrix_layout: MatrixLayout

tile_size: TileSize

plane_dim: u32

Implementations

impl<A: Numeric> InterleavedTileExpand<A>

pub fn __expand_mma_method<L: Numeric, R: Numeric>( &mut self, scope: &Scope, 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.

pub fn __expand_mma<L: Numeric, R: Numeric>( scope: &Scope, this: &mut Self, 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> InterleavedTileExpand<N>

pub fn __expand_copy_from_method<SE: Numeric, SS: Size, Sc: TileScope>( &mut self, scope: &Scope, 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_method(&mut self, scope: &Scope)

pub fn __expand_copy_from<SE: Numeric, SS: Size, Sc: TileScope>( scope: &Scope, this: &mut Self, 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)

Trait Implementations

impl<N: Numeric> AsMutExpand for InterleavedTileExpand<N>

fn __expand_ref_mut_method(&mut self, _: &Scope) -> &mut Self

fn __expand_as_mut_method(&mut self, scope: &Scope) -> &mut T

impl<N: Numeric> AsRefExpand for InterleavedTileExpand<N>

fn __expand_ref_method(&self, _: &Scope) -> &Self

fn __expand_as_ref_method(&self, scope: &Scope) -> &T

impl<N: Numeric> CubeDebug for InterleavedTileExpand<N>

fn set_debug_name(&self, scope: &Scope, name: &'static str)

Set the debug name of this type’s expansion. Should do nothing for types that don’t appear at runtime

impl<N: Numeric> ExpandTypeClone for InterleavedTileExpand<N>

fn clone_unchecked(&self) -> Self

Unchecked clone that only clones the conceptual runtime value. Should only be used in cases where each copy is used in a mutually exclusive branch (i.e. match, runtime enums). This is intentionally separated from Rust’s Clone semantics and should only be used for the conceptual expand values, never real data. Using two values in the same branch is undefined behaviour.

impl<N: Numeric> IntoExpand for InterleavedTileExpand<N>

type Expand = InterleavedTileExpand<N>

fn into_expand(self, _: &Scope) -> Self

impl<N: Numeric> IntoMut for InterleavedTileExpand<N>

fn into_mut(self, scope: &Scope) -> Self

Convert the variable into a potentially new mutable variable in scope, copying if needed.