Struct Tile

pub struct Tile<ES: Numeric> {
    pub slice: Slice<Line<ES>>,
    pub stride: u32,
}

Data to be handed to the tile matmul

Fields

slice: Slice<Line<ES>>

Slice containing all data

stride: u32

Stride between each row/col, depending on MatrixLayout (the other is assumed to be 1)

Implementations

impl<ES: Numeric> Tile<ES>

pub fn new_contiguous<T: TileConfig>( slice: Slice<Line<ES>>, ident: Ident, config: T, ) -> Tile<ES>

pub fn new_strided(slice: Slice<Line<ES>>, stride: u32) -> Tile<ES>

pub fn as_unlined<T: TileConfig>( &self, ident: Ident, config: T, ) -> (Slice<ES>, u32)

pub fn __expand_new_contiguous<T: TileConfig>( context: &mut Scope, slice: <Slice<Line<ES>> as CubeType>::ExpandType, ident: Ident, config: T, ) -> <Tile<ES> as CubeType>::ExpandType

pub fn __expand_new_strided( context: &mut Scope, slice: <Slice<Line<ES>> as CubeType>::ExpandType, stride: <u32 as CubeType>::ExpandType, ) -> <Tile<ES> as CubeType>::ExpandType

pub fn __expand_as_unlined<T: TileConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, ident: Ident, config: T, ) -> <(Slice<ES>, u32) as CubeType>::ExpandType

Trait Implementations

impl<ES: Numeric> CubeType for Tile<ES>

type ExpandType = TileExpand<ES>

fn init(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.