Struct ErasedReducePlan 

pub struct ErasedReducePlan { /* private fields */ }

Dtype-erased reduction wrapper.

This is the erased replay boundary for full-tensor scalar reductions and axis reductions with a fixed output layout. It supports only operations with an unambiguous identity value in the selected dtype.

Implementations

impl ErasedReducePlan

pub fn compile( dtype: KernelDType, op: ReduceOp, dims: &[usize], src_strides: &[isize], ) -> Result<ErasedReducePlan, StridedError>

Validate and store a full-reduction plan for one dtype and source layout.

pub fn compile_axes( dtype: KernelDType, op: ReduceOp, src_dims: &[usize], src_strides: &[isize], dest_dims: &[usize], dest_strides: &[isize], axes: &[usize], ) -> Result<ErasedReducePlan, StridedError>

Validate and store an axis-reduction plan for one dtype and fixed source/output layouts.

axes names the source axes reduced away. Output dimensions must be the remaining source dimensions in source-axis order. When all axes are reduced, any output layout with exactly one reachable element is accepted.

pub fn dtype(&self) -> KernelDType

pub fn op(&self) -> ReduceOp

pub fn execute( &self, ctx: &ExecContext, dest: &mut ErasedRawStridedMut<'_>, src: &ErasedRawStridedRef<'_>, ) -> Result<(), StridedError>

Execute the reduction into an erased output descriptor.

pub fn execute_uninit( &self, ctx: &ExecContext, dest: &mut ErasedRawStridedUninitMut<'_>, src: &ErasedRawStridedPtr<'_>, ) -> Result<(), StridedError>

On success, every reachable destination slot is fully overwritten; unreachable holes are neither read nor initialized. Validation errors are returned before any destination write. A panic during execution may leave a partially initialized MaybeUninit destination, which is still safely droppable; no readable value is promised for unwritten reachable slots.

Trait Implementations

impl Clone for ErasedReducePlan

fn clone(&self) -> ErasedReducePlan

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ErasedReducePlan

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

Formats the value using the given formatter.