Struct PlaneWriter

pub struct PlaneWriter<EG: Numeric> {
    pub tensor_writer: TensorWriter<EG>,
}

Writes tiles from out shared memory to output global memory using a plane for each tile

Fields

tensor_writer: TensorWriter<EG>

Implementations

impl<EG: Numeric> PlaneWriter<EG>

pub fn new( tensor: VirtualTensor<EG, ReadWrite>, x_offset: u32, y_offset: u32, batch_offset: u32, ) -> Self

pub fn __expand_new( scope: &mut Scope, tensor: <VirtualTensor<EG, ReadWrite> as CubeType>::ExpandType, x_offset: <u32 as CubeType>::ExpandType, y_offset: <u32 as CubeType>::ExpandType, batch_offset: <u32 as CubeType>::ExpandType, ) -> <Self as CubeType>::ExpandType

Trait Implementations

impl<EG: Numeric> CubeType for PlaneWriter<EG>

type ExpandType = PlaneWriterExpand<EG>

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

Wrapper around the init method, necessary to type inference.

impl<EG: Numeric> GlobalWriter<EG> for PlaneWriter<EG>

fn write<G: GlobalConfig>( this: &mut Self, out_smem_slice: Slice<Line<EG>>, tile_row: u32, tile_col: u32, config: G, )

Writes the given slice to global memory, at a position that depends on plane and accumulator indexes.

fn __expand_write<G: GlobalConfig>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, out_smem_slice: <Slice<Line<EG>> as CubeType>::ExpandType, tile_row: <u32 as CubeType>::ExpandType, tile_col: <u32 as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType