Struct ArgAccumulator

pub struct ArgAccumulator<N: Numeric> {
    pub elements: SharedMemory<Line<N>>,
    pub args: SharedMemory<Line<u32>>,
}

A pair of shared memory used for [ArgMax] and [ArgMin].

Fields

elements: SharedMemory<Line<N>>

args: SharedMemory<Line<u32>>

Trait Implementations

impl<N: Numeric> CubeType for ArgAccumulator<N>

type ExpandType = ArgAccumulatorExpand<N>

fn init(context: &mut CubeContext, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl<N: Numeric> IntoRuntime for ArgAccumulator<N>

fn __expand_runtime_method(self, context: &mut CubeContext) -> Self::ExpandType

fn runtime(self) -> Self

Make sure a type is actually expanded into its runtime expand type.

impl<In: Numeric> SharedAccumulator<In> for ArgAccumulator<In>

type Item = (Line<In>, Line<u32>)

fn allocate(length: u32, line_size: u32) -> Self

fn read(accumulator: &Self, index: u32) -> Self::Item

fn write(accumulator: &mut Self, index: u32, item: Self::Item)

fn __expand_allocate( context: &mut CubeContext, length: u32, line_size: u32, ) -> <Self as CubeType>::ExpandType

fn __expand_read( context: &mut CubeContext, accumulator: <Self as CubeType>::ExpandType, index: <u32 as CubeType>::ExpandType, ) -> <Self::Item as CubeType>::ExpandType

fn __expand_write( context: &mut CubeContext, accumulator: <Self as CubeType>::ExpandType, index: <u32 as CubeType>::ExpandType, item: <Self::Item as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType