Struct TmaIm2colGlobalReaderExpand 

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

Fields

map: <Im2colTmaReader<IP::Global> as CubeType>::ExpandType

stages: <Sequence<StridedStage<IP::Stage, TmaIm2colTiling>> as CubeType>::ExpandType

Implementations

impl<IP: MatrixPrecision> TmaIm2colGlobalReaderExpand<IP>

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

pub fn __expand_advance_view_method( self, scope: &mut Scope, k_offset: <u32 as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

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

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

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

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

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

Trait Implementations

impl<IP: MatrixPrecision> Clone for TmaIm2colGlobalReaderExpand<IP>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<IP: MatrixPrecision> CubeDebug for TmaIm2colGlobalReaderExpand<IP>

fn set_debug_name(&self, scope: &mut Scope, name: &'static str)

Set the debug name of this type’s expansion. Should do nothing for types that don’t appear at runtime

impl<IP: MatrixPrecision> IntoMut for TmaIm2colGlobalReaderExpand<IP>

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