Struct Sum

pub struct Sum;

Trait Implementations

impl Debug for Sum

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Reduce for Sum

type Instruction<In: Numeric> = Sum

impl<In> ReduceInstruction<In> for Sum

where In: Numeric,

type AccumulatorItem = Line<In>

The intermediate state into which we accumulate new input elements. This is most likely a Line<T> or a struct or tuple of lines.

type SharedAccumulator = SharedMemory<Line<In>>

When multiple agents are collaborating to reduce a single slice, we need a share accumulator to store multiple AccumulatorItem. This is most likely a SharedMemory<Line<T>> or a struct or tuple of lined shared memories.

fn null_input(line_size: u32) -> Line<In>

A input such that Self::reduce(accumulator, Self::null_input(), coordinate, use_planes) is guaranteed to return accumulator unchanged for any choice of coordinate.

fn null_accumulator( line_size: u32, ) -> <Sum as ReduceInstruction<In>>::AccumulatorItem

A accumulator such that Self::fuse_accumulators(accumulator, Self::null_accumulator() always returns is guaranteed to return accumulator unchanged.

fn assign_accumulator( destination: &mut <Sum as ReduceInstruction<In>>::AccumulatorItem, source: &<Sum as ReduceInstruction<In>>::AccumulatorItem, )

Assign the value of source into destination. In spirit, this is equivalent to destination = source;, but this syntax is not currently supported by CubeCL.

fn reduce( accumulator: &<Sum as ReduceInstruction<In>>::AccumulatorItem, item: Line<In>, _coordinate: Line<u32>, use_planes: bool, ) -> <Sum as ReduceInstruction<In>>::AccumulatorItem

If use_planes is true, reduce all the item and coordinate within the accumulator. Else, reduce the given item and coordinate into the accumulator.

fn fuse_accumulators( lhs: <Sum as ReduceInstruction<In>>::AccumulatorItem, rhs: <Sum as ReduceInstruction<In>>::AccumulatorItem, ) -> <Sum as ReduceInstruction<In>>::AccumulatorItem

Reduce two accumulators into a single accumulator.

fn merge_line<Out>( accumulator: <Sum as ReduceInstruction<In>>::AccumulatorItem, _shape_axis_reduce: u32, ) -> Out

where Out: Numeric,

Reduce all elements of the accumulator into a single output element of type Out.

fn to_output_perpendicular<Out>( accumulator: <Sum as ReduceInstruction<In>>::AccumulatorItem, _shape_axis_reduce: u32, ) -> Line<Out>

where Out: Numeric,

Convert each element of the accumulator into the expected output element of type Out.

fn __expand_null_input( context: &mut CubeContext, line_size: u32, ) -> <Line<In> as CubeType>::ExpandType

fn __expand_null_accumulator( context: &mut CubeContext, line_size: u32, ) -> <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType

fn __expand_assign_accumulator( context: &mut CubeContext, destination: <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType, source: <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

fn __expand_reduce( context: &mut CubeContext, accumulator: <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType, item: <Line<In> as CubeType>::ExpandType, _coordinate: <Line<u32> as CubeType>::ExpandType, use_planes: bool, ) -> <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType

fn __expand_fuse_accumulators( context: &mut CubeContext, lhs: <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType, rhs: <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType, ) -> <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType

fn __expand_merge_line<Out>( context: &mut CubeContext, accumulator: <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType, _shape_axis_reduce: <u32 as CubeType>::ExpandType, ) -> <Out as CubeType>::ExpandType

where Out: Numeric,

fn __expand_to_output_perpendicular<Out>( context: &mut CubeContext, accumulator: <<Sum as ReduceInstruction<In>>::AccumulatorItem as CubeType>::ExpandType, _shape_axis_reduce: <u32 as CubeType>::ExpandType, ) -> <Line<Out> as CubeType>::ExpandType

where Out: Numeric,