Struct TmaIm2colGlobalReader 

pub struct TmaIm2colGlobalReader<IP: MatrixPrecision> {
    pub map: Im2colTmaReader<IP::Global>,
    pub stages: Sequence<StridedStage<IP::Stage, TmaIm2colTiling>>,
    /* private fields */
}

Reader that translates matrix coordinates to input coordinates using the im2col algorithm

Fields

map: Im2colTmaReader<IP::Global>

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

Implementations

impl<IP: MatrixPrecision> TmaIm2colGlobalReader<IP>

pub fn new( tensor: TensorMap<Line<IP::Global>>, x_offset: u32, y_offset: u32, runtime_args: &RuntimeArgs, num_stages: u32, params: ConvolutionParams, config: StageMemoryConfig, ) -> Self

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

pub fn advance_view(&mut self, k_offset: u32)

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

pub fn __expand_new( scope: &mut Scope, tensor: <TensorMap<Line<IP::Global>> as CubeType>::ExpandType, x_offset: <u32 as CubeType>::ExpandType, y_offset: <u32 as CubeType>::ExpandType, runtime_args: <RuntimeArgs as CubeType>::ExpandType, num_stages: u32, params: ConvolutionParams, config: StageMemoryConfig, ) -> <Self as CubeType>::ExpandType

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

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

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

Trait Implementations

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

type ExpandType = TmaIm2colGlobalReaderExpand<IP>

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

Wrapper around the init method, necessary to type inference.