Struct TmaWeightGlobalReader 

pub struct TmaWeightGlobalReader<IP: MatrixPrecision> {
    pub global_iter: GlobalIterator<Line<IP::Global>>,
    pub stages: Sequence<StridedStage<IP::Stage, TmaWeightTiling>>,
    /* private fields */
}

Fields

global_iter: GlobalIterator<Line<IP::Global>>

stages: Sequence<StridedStage<IP::Stage, TmaWeightTiling>>

Implementations

impl<IP: MatrixPrecision> TmaWeightGlobalReader<IP>

pub fn new( global_view: View<Line<IP::Global>, Coords2d>, k_step: u32, num_stages: u32, config: StageMemoryConfig, ) -> Self

pub fn fill_stage(&mut self, barrier: &Barrier, stage_idx: u32)

pub fn stage(&self, stage_idx: u32) -> TmaWeightStage<IP>

pub fn advance_view(&mut self)

pub fn __expand_new( scope: &mut Scope, global_view: <View<Line<IP::Global>, Coords2d> as CubeType>::ExpandType, k_step: <u32 as CubeType>::ExpandType, num_stages: u32, config: StageMemoryConfig, ) -> <Self as CubeType>::ExpandType

pub fn __expand_fill_stage( scope: &mut Scope, this: <Self as CubeType>::ExpandType, barrier: <Barrier as CubeType>::ExpandType, stage_idx: u32, ) -> <() as CubeType>::ExpandType

pub fn __expand_stage( scope: &mut Scope, this: <Self as CubeType>::ExpandType, stage_idx: u32, ) -> <TmaWeightStage<IP> as CubeType>::ExpandType

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

Trait Implementations

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

type ExpandType = TmaWeightGlobalReaderExpand<IP>

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

Wrapper around the init method, necessary to type inference.