Struct PlaneVecTile 

pub struct PlaneVecTile<N: Numeric> {
    pub data: Array<Vector<N, NPlaneVec>>,
    pub matrix_layout: MatrixLayout,
    pub tile_size: TileSize,
    pub reduce_vector_size: u32,
}

Plane-vec tile. Holds the per-unit fragment plus the minimal comptime data the tile body actually uses (tile_size for the n iteration, plus the register-size hookup info implicit in the NPlaneVec binding done at allocation time). The matmul-level configuration that produced these values lives in cubek-matmul as PlaneVecMatInnerProduct.

Fields

data: Array<Vector<N, NPlaneVec>>

matrix_layout: MatrixLayout

tile_size: TileSize

reduce_vector_size: u32

Inner reduction vector size for NPlaneVec. Carried because planevec_write_to_shared needs the extent at use-site (NPlaneVec::value() isn’t observable from a #[cube] callsite).

Implementations

impl<A: Numeric> PlaneVecTile<A>

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

Executes lhs · rhs, accumulating into self via the plane-vec inner-product matmul.

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

Executes lhs · rhs, accumulating into self via the plane-vec inner-product matmul.

impl<N: Numeric> PlaneVecTile<N>

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

Copies into the plane-vec 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 plane-vec 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 PlaneVecTile<N>

type ExpandType = PlaneVecTileExpand<N>