Struct ErasedScatterPlan 

pub struct ErasedScatterPlan { /* private fields */ }

Dtype-erased additive scatter wrapper.

Implementations

impl ErasedScatterPlan

pub fn compile( dtype: KernelDType, index_dtype: KernelDType, operand_dims: &[usize], operand_strides: &[isize], index_dims: &[usize], index_strides: &[isize], update_dims: &[usize], update_strides: &[isize], dest_dims: &[usize], dest_strides: &[isize], spec: ScatterSpec, ) -> Result<Self>

Validate and store an additive scatter plan for one value dtype, index dtype, and layout set.

pub fn dtype(&self) -> KernelDType

pub fn index_dtype(&self) -> KernelDType

pub fn plan(&self) -> &ScatterPlan

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

Execute additive scatter into an erased output descriptor.

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

On success, the copy phase initializes every reachable destination slot before the read-modify-write phase. Unreachable holes are neither read nor initialized. Validation errors before the copy leave the destination untouched; an error or panic after the copy may leave a mixture of old and new reachable values, all initialized and safely droppable.

Trait Implementations

impl Clone for ErasedScatterPlan

fn clone(&self) -> ErasedScatterPlan

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ErasedScatterPlan

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

Formats the value using the given formatter.