Struct AccumulateArg 

pub struct AccumulateArg<'a, E: Numeric> {
    pub tensor: &'a Tensor<Atomic<E>>,
    pub width: usize,
    pub spec: TileSpec,
}

An output several instances accumulate into, as a single launch argument, in lines of width. The buffer must arrive holding zero: the atomic only ever adds.

Fields

tensor: &'a Tensor<Atomic<E>>

width: usize

The width a line is added in: the buffer’s own elements are scalar.

spec: TileSpec

Implementations

impl<'a, E: Numeric> AccumulateArg<'a, E>

pub fn tile(&self, partitioning: &Partitioning) -> Tile

Serve the output as a Tile that accumulates into it in lines of its width.

pub fn __expand_tile( scope: &Scope, this: &<Self as CubeType>::ExpandType, partitioning: &<Partitioning as CubeType>::ExpandType, ) -> <Tile as CubeType>::ExpandType

Serve the output as a Tile that accumulates into it in lines of its width.

Trait Implementations

impl<'a, E: Numeric> CubeType for AccumulateArg<'a, E>

type ExpandType = AccumulateArgExpand<'a, E>

impl<'a, E: Numeric> LaunchArg for AccumulateArg<'a, E>

where &'a Tensor<Atomic<E>>: LaunchArg,

type RuntimeArg = AccumulateArgLaunch<'a, E>

The runtime argument for the kernel.

type CompilationArg = AccumulateArgCompilationArg<'a, E>

Compilation argument.

fn register( arg: Self::RuntimeArg, launcher: &mut KernelLauncher, ) -> Self::CompilationArg

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register a variable during compilation that fill the KernelBuilder.