Struct CmmaTile 

pub struct CmmaTile<N: Numeric> {
    pub matrix: Matrix<N>,
    pub matrix_layout: MatrixLayout,
    pub tile_size: TileSize,
}

Fields

matrix: Matrix<N>

matrix_layout: MatrixLayout

tile_size: TileSize

Implementations

impl<E: Float> CmmaTile<E>

pub fn fill_zero(&mut self)

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

impl<A: Numeric> CmmaTile<A>

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

Executes lhs · rhs, accumulating into self.

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

Executes lhs · rhs, accumulating into self.

impl<N: Numeric> CmmaTile<N>

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

Supported sources: SharedTile (load) 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, )

Supported sources: SharedTile (load) and None (zero-init).

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

Trait Implementations

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

type ExpandType = CmmaTileExpand<N>