Struct GlobalIterator 

pub struct GlobalIterator<EI: CubePrimitive> { /* private fields */ }

An iterator over global memory, advancing along k.

Implementations

impl<EG: CubePrimitive> GlobalIterator<EG>

pub fn new( global_view: View<EG, Coords2d>, step: u32, view_direction: ViewDirection, checked: bool, ) -> Self

Instantiate a read iterator over the given global view, which should be sliced to the size of one m/n stage and the full range of k handled by this matmul instance.

step is the amount advanced in view_direction each iteration. checked determines whether the slices should be created as checked or unchecked.

pub fn advance(&self)

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

pub fn view(&self) -> View<EG, Coords2d>

Returns the current view slice of the iterator

pub fn line_size(&self) -> u32

Returns the line size of the global view

pub fn __expand_new( scope: &mut Scope, global_view: <View<EG, Coords2d> as CubeType>::ExpandType, step: <u32 as CubeType>::ExpandType, view_direction: ViewDirection, checked: bool, ) -> <Self as CubeType>::ExpandType

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

pub fn __expand_view( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <View<EG, Coords2d> as CubeType>::ExpandType

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

Trait Implementations

impl<EI: Clone + CubePrimitive> Clone for GlobalIterator<EI>

fn clone(&self) -> GlobalIterator<EI>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<EI: CubePrimitive> CubeType for GlobalIterator<EI>

type ExpandType = GlobalIteratorExpand<EI>

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

Wrapper around the init method, necessary to type inference.

impl<EG: CubePrimitive> Send for GlobalIterator<EG>

impl<EG: CubePrimitive> Sync for GlobalIterator<EG>