Enum BiasGlobalReader 

pub enum BiasGlobalReader<IP: MatrixPrecision> {
    Some {
        view: View<Line<IP::Global>, Coords2d>,
        stage: StridedStage<IP::Stage, BiasTilingLayout>,
    },
    None,
}

Special reader to broadcast the 1D bias to the 2D accumulator matrix

Variants

Some

Fields

view: View<Line<IP::Global>, Coords2d>

stage: StridedStage<IP::Stage, BiasTilingLayout>

None

Implementations

impl<IP: MatrixPrecision> BiasGlobalReader<IP>

pub fn new_Some( view: View<Line<IP::Global>, Coords2d>, stage: StridedStage<IP::Stage, BiasTilingLayout>, ) -> Self

pub fn __expand_new_Some( _: &mut Scope, view: <View<Line<IP::Global>, Coords2d> as CubeType>::ExpandType, stage: <StridedStage<IP::Stage, BiasTilingLayout> as CubeType>::ExpandType, ) -> BiasGlobalReaderExpand<IP>

pub fn new_None() -> Self

pub fn __expand_new_None(_: &mut Scope) -> BiasGlobalReaderExpand<IP>

impl<IP: MatrixPrecision> BiasGlobalReader<IP>

pub fn load_stage<G: GlobalConfig>(&mut self, config: G)

Reads all bias tiles into the stage. Unlike normal readers, bias only reads a 1D vector along the n dimension.

pub fn stage(&self) -> BiasStage<IP::Stage>

Return the stage contained in this global reader. It will use custom tiling with a stride of 0.

pub fn __expand_load_stage<G: GlobalConfig>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

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

impl<IP: MatrixPrecision> BiasGlobalReader<IP>

pub fn new( view: CubeOption<View<Line<IP::Global>, Coords2d>>, config: StageMemoryConfig, ) -> Self

Create a new bias reader from the bias tensor and a global offset n_offset.

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

Trait Implementations

impl<IP: MatrixPrecision> CubeDebug for BiasGlobalReader<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> CubeType for BiasGlobalReader<IP>

type ExpandType = BiasGlobalReaderExpand<IP>

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

Wrapper around the init method, necessary to type inference.