Struct MmaTile 

pub struct MmaTile<N: Numeric> {
    pub fragment: MmaFragment<N>,
    pub matrix_layout: MatrixLayout,
    pub tile_size: TileSize,
    pub mma_io_config: MmaIOConfig,
}

Single MMA tile carrier. The role (Lhs / Rhs / Acc) lives inside MmaFragment because each role’s fragment uses a different inner vector size (NL / NR / NA); the outer carrier holds the shared comptime metadata the tile body uses. The matmul-level configuration that produced these values lives in cubek-matmul as MmaMatmul.

Fields

fragment: MmaFragment<N>

matrix_layout: MatrixLayout

tile_size: TileSize

mma_io_config: MmaIOConfig

Implementations

impl<A: Numeric> MmaTile<A>

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

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 as CubeType>::ExpandType, 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> MmaTile<N>

pub fn copy_from<SE: Numeric, SS: Size, L: Numeric, R: Numeric, A: Numeric, Sc: TileScope>( &mut self, source: &Tile<SE, Sc>, _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 init_zero<L: Numeric, R: Numeric>(&mut self)

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 as CubeType>::ExpandType, 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 as CubeType>::ExpandType, )

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

Trait Implementations

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

type ExpandType = MmaTileExpand<N>