Enum BiasLoader

pub enum BiasLoader<MP: MatmulPrecision> {
    Some {
        tensor_view: BiasReader<MP::EO>,
        stage: Stage<MP::EA, ConvTilingLayout>,
    },
    None,
}

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

Variants

Some

Fields

tensor_view: BiasReader<MP::EO>

stage: Stage<MP::EA, ConvTilingLayout>

None

Implementations

impl<MP: MatmulPrecision> BiasLoader<MP>

pub fn new_Some( tensor_view: BiasReader<MP::EO>, stage: Stage<MP::EA, ConvTilingLayout>, ) -> Self

pub fn __expand_new_Some( _: &mut Scope, tensor_view: <BiasReader<MP::EO> as CubeType>::ExpandType, stage: <Stage<MP::EA, ConvTilingLayout> as CubeType>::ExpandType, ) -> BiasLoaderExpand<MP>

pub fn new_None() -> Self

pub fn __expand_new_None(_: &mut Scope) -> BiasLoaderExpand<MP>

impl<MP: MatmulPrecision> BiasLoader<MP>

pub fn new<G: GlobalConfig>( tensor: CubeOption<VirtualTensor<MP::EO>>, n_offset: u32, config: G, ) -> Self

pub fn __expand_new<G: GlobalConfig>( context: &mut Scope, tensor: <CubeOption<VirtualTensor<MP::EO>> as CubeType>::ExpandType, n_offset: <u32 as CubeType>::ExpandType, config: G, ) -> <Self as CubeType>::ExpandType

Trait Implementations

impl<MP: MatmulPrecision> AccumulatorLoader<MP> for BiasLoader<MP>

fn load<TMM: TileMatmul<MP>>( this: &mut Self, acc: &mut TMM::Accumulator, tile_n: u32, config: TMM::Config, )

Load accumulator

fn fill_stage<G: GlobalConfig>(this: &mut Self, config: G)

fn __expand_fill_stage<G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_load<TMM: TileMatmul<MP>>( context: &mut Scope, this: <Self as CubeType>::ExpandType, acc: <TMM::Accumulator as CubeType>::ExpandType, tile_n: <u32 as CubeType>::ExpandType, config: TMM::Config, ) -> <() as CubeType>::ExpandType

impl<MP: MatmulPrecision> CubeDebug for BiasLoader<MP>

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<MP: MatmulPrecision> CubeType for BiasLoader<MP>

type ExpandType = BiasLoaderExpand<MP>

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

Wrapper around the init method, necessary to type inference.