Struct ErasedGatherPlan 

pub struct ErasedGatherPlan { /* private fields */ }

Dtype-erased gather wrapper.

This is the erased replay boundary for indexed reads. Value buffers use the configured value dtype, while the index descriptor must use i32 or i64.

Implementations

impl ErasedGatherPlan

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

Validate and store a gather 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) -> &GatherPlan

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

Execute an indexed read into an erased output descriptor.

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

Execute gather into a destination whose reachable slots may be uninitialized. All validation precedes the first destination write. 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 ErasedGatherPlan

fn clone(&self) -> ErasedGatherPlan

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ErasedGatherPlan

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

Formats the value using the given formatter.