Struct MmaTileExpand 

pub struct MmaTileExpand<N: Numeric> {
    pub fragment: <MmaFragment<N> as CubeType>::ExpandType,
    pub matrix_layout: MatrixLayout,
    pub tile_size: TileSize,
    pub mma_io_config: MmaIOConfig,
}

Fields

fragment: <MmaFragment<N> as CubeType>::ExpandType

matrix_layout: MatrixLayout

tile_size: TileSize

mma_io_config: MmaIOConfig

Implementations

impl<A: Numeric> MmaTileExpand<A>

pub fn __expand_mma_method<L: Numeric, R: Numeric>( &mut self, scope: &Scope, lhs: &<MmaTile<L> as CubeType>::ExpandType, rhs: &<MmaTile<R> as CubeType>::ExpandType, )

Executes lhs · rhs, accumulating into self. Each operand must carry the role its position requires (Lhs, Rhs, Acc).

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

Executes lhs · rhs, accumulating into self. Each operand must carry the role its position requires (Lhs, Rhs, Acc).

impl<N: Numeric> MmaTileExpand<N>

pub fn __expand_copy_from_method<SE: Numeric, SS: Size, L: Numeric, R: Numeric, A: Numeric, Sc: TileScope>( &mut self, scope: &Scope, source: &<Tile<SE, Sc> as CubeType>::ExpandType, _ident: StageIdent, )

Copies into the mma fragment from source. Supported sources: Shared (per-role load) and None (zero-init, Acc only). L / R / A are the matmul triple’s role types — needed by the per-role load functions. When self is in role X, N substitutes for the X type and the other two are taken from the caller’s generics.

pub fn __expand_init_zero_method<L: Numeric, R: Numeric>( &mut self, scope: &Scope, )

Zero-init the mma fragment (Acc role only).

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

Copies into the mma fragment from source. Supported sources: Shared (per-role load) and None (zero-init, Acc only). L / R / A are the matmul triple’s role types — needed by the per-role load functions. When self is in role X, N substitutes for the X type and the other two are taken from the caller’s generics.

pub fn __expand_init_zero<L: Numeric, R: Numeric>( scope: &Scope, this: &mut Self, )

Zero-init the mma fragment (Acc role only).

Trait Implementations

impl<N: Numeric> AsMutExpand for MmaTileExpand<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 MmaTileExpand<N>

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

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

impl<N: Numeric> CubeDebug for MmaTileExpand<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 MmaTileExpand<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 MmaTileExpand<N>

type Expand = MmaTileExpand<N>

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

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

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

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