Struct PlaneWriter 

pub struct PlaneWriter<IP: MatrixPrecision> { /* private fields */ }

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

Implementations

impl<IP: MatrixPrecision> PlaneWriter<IP>

pub fn new<S: StageConfig>( global: View<Line<IP::Global>, Coords2d, ReadWrite>, global_config: GlobalMemoryConfig, stage_config: S, ) -> Self

pub fn __expand_new<S: StageConfig>( scope: &mut Scope, global: <View<Line<IP::Global>, Coords2d, ReadWrite> as CubeType>::ExpandType, global_config: GlobalMemoryConfig, stage_config: S, ) -> <Self as CubeType>::ExpandType

Trait Implementations

impl<IP: MatrixPrecision> CubeType for PlaneWriter<IP>

type ExpandType = PlaneWriterExpand<IP>

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

Wrapper around the init method, necessary to type inference.

impl<IP: MatrixPrecision> GlobalWriter<IP> for PlaneWriter<IP>

type Stage = PartitionedStage<<IP as MatrixPrecision>::Stage>

Tile stage that stores the data for this writer

fn init<S: StageConfig>( tensor: View<Line<IP::Global>, Coords2d, ReadWrite>, config: GlobalMemoryConfig, stage_config: S, ) -> Self

Init this writer from a global tensor and config

fn stage(this: &Self) -> Self::Stage

Stage used by this writer

fn __expand_init<S: StageConfig>( scope: &mut Scope, tensor: <View<Line<IP::Global>, Coords2d, ReadWrite> as CubeType>::ExpandType, config: GlobalMemoryConfig, stage_config: S, ) -> <Self as CubeType>::ExpandType

fn __expand_stage( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <Self::Stage as CubeType>::ExpandType

impl<IP: MatrixPrecision> WriteEventListener for PlaneWriter<IP>

fn on_event(this: &mut Self, event: WriteEvent)

fn __expand_on_event( scope: &mut Scope, this: <Self as CubeType>::ExpandType, event: <WriteEvent as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType