Struct ZeroAccumulatorLoader

pub struct ZeroAccumulatorLoader;

Accumulator loader that zeros the accumulator

Trait Implementations

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

fn fill_stage<G: GlobalConfig>(_this: &mut Self, _config: G)

fn load<Tile: TileMatmul<MP>>( _this: &mut Self, acc: &mut Tile::Accumulator, _n_tile: u32, config: Tile::Config, )

Load accumulator for nth_tile. Should call either zero_accumulator or fill_accumulator for the underlying tile.

fn __expand_fill_stage<G: GlobalConfig>( scope: &mut Scope, _this: <Self as CubeType>::ExpandType, _config: G, ) -> <() as CubeType>::ExpandType

fn __expand_load<Tile: TileMatmul<MP>>( scope: &mut Scope, _this: <Self as CubeType>::ExpandType, acc: <Tile::Accumulator as CubeType>::ExpandType, _n_tile: <u32 as CubeType>::ExpandType, config: Tile::Config, ) -> <() as CubeType>::ExpandType

impl CubeType for ZeroAccumulatorLoader

type ExpandType = ZeroAccumulatorLoaderExpand

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

Wrapper around the init method, necessary to type inference.