Struct SyncFullStageGlobalReader 

pub struct SyncFullStageGlobalReader<IP: MatrixPrecision, G: GlobalConfig, L: SyncFullLoadingStrategy> { /* private fields */ }

Loads the entire stage memory using synchronous data movement operations.

A complete load is referred to as a Job, which is divided into Tasks— each Task represents a single data transfer for a specific unit

Implementations

impl<IP: MatrixPrecision, G: GlobalConfig, L: SyncFullLoadingStrategy> SyncFullStageGlobalReader<IP, G, L>

pub fn new( tensor: View<Line<IP::Global>, Coords2d>, k_step: u32, ident: MatmulIdent, config: G, ) -> Self

Create a new SyncFullStageGlobalReader

pub fn stage(&self) -> StridedStage<IP::Stage, L::TilingLayout>

Give a reader to the loaded stage memory.

pub fn free_stage(self)

pub fn advance_view(&mut self)

Advance the view over global memory along the k dimension by a specified offset, k_offset.

pub fn load_stage(&mut self, config: G)

Accomplish the entire job of loading data into the stage memory

pub fn __expand_new( scope: &mut Scope, tensor: <View<Line<IP::Global>, Coords2d> as CubeType>::ExpandType, k_step: <u32 as CubeType>::ExpandType, ident: MatmulIdent, config: G, ) -> <Self as CubeType>::ExpandType

pub fn __expand_stage( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <StridedStage<IP::Stage, L::TilingLayout> as CubeType>::ExpandType

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

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

pub fn __expand_load_stage( scope: &mut Scope, this: <Self as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

Trait Implementations

impl<IP: Clone + MatrixPrecision, G: Clone + GlobalConfig, L: Clone + SyncFullLoadingStrategy> Clone for SyncFullStageGlobalReader<IP, G, L>

where IP::Global: Clone, IP::Stage: Clone, L::TilingLayout: Clone, L::Job<IP>: Clone,

fn clone(&self) -> SyncFullStageGlobalReader<IP, G, L>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<IP: MatrixPrecision, G: GlobalConfig, L: SyncFullLoadingStrategy> CubeType for SyncFullStageGlobalReader<IP, G, L>

type ExpandType = SyncFullStageGlobalReaderExpand<IP, G, L>

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

Wrapper around the init method, necessary to type inference.

impl<IP: MatrixPrecision, G: GlobalConfig, L: SyncFullLoadingStrategy> JobExecutor<G> for SyncFullStageGlobalReader<IP, G, L>

type JobIterator = SyncFullStageJobIterator<IP, L>

The job to execute

fn create_job_iterator( this: &Self, _stage_buffer: StageBuffer, config: G, ) -> Self::JobIterator

Create the job to execute

fn execute_task( this: &mut Self, job_iterator: &mut SyncFullStageJobIterator<IP, L>, config: G, )

Execute the next task

fn execute_all_remaining_tasks( this: &mut Self, job_iterator: &mut Self::JobIterator, config: G, )

Execute all tasks that remain at once

fn execute_whole_job(this: &mut Self, stage_buffer: StageBuffer, config: G)

Create a job and execute all its tasks at once

fn __expand_create_job_iterator( scope: &mut Scope, this: <Self as CubeType>::ExpandType, _stage_buffer: StageBuffer, config: G, ) -> <Self::JobIterator as CubeType>::ExpandType

fn __expand_execute_task( scope: &mut Scope, this: <Self as CubeType>::ExpandType, job_iterator: <SyncFullStageJobIterator<IP, L> as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_execute_all_remaining_tasks( scope: &mut Scope, this: <Self as CubeType>::ExpandType, job_iterator: <Self::JobIterator as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_execute_whole_job( scope: &mut Scope, this: <Self as CubeType>::ExpandType, stage_buffer: StageBuffer, config: G, ) -> <() as CubeType>::ExpandType