Struct SyncFullLoaderJobIterator

pub struct SyncFullLoaderJobIterator<MP: MatmulPrecision, L: SyncFullLoadingStrategy> {
    pub num_tasks: u32,
    pub current: ComptimeCell<TaskCounter>,
    /* private fields */
}

A comptime iterator over a job for sync full loader

Fields

num_tasks: u32

current: ComptimeCell<TaskCounter>

Trait Implementations

impl<MP: MatmulPrecision, L: SyncFullLoadingStrategy> CubeType for SyncFullLoaderJobIterator<MP, L>

type ExpandType = SyncFullLoaderJobIteratorExpand<MP, L>

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

Wrapper around the init method, necessary to type inference.

impl<MP: MatmulPrecision, L: SyncFullLoadingStrategy> JobIterator for SyncFullLoaderJobIterator<MP, L>

fn current(this: &Self) -> u32

Get the index of the current task

fn num_tasks(this: &Self) -> u32

Get the number of tasks

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

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