Struct ScanKernel 

pub struct ScanKernel { /* private fields */ }

Prefix sum (scan) operation

Implementations

impl ScanKernel

pub fn inclusive() -> Self

Create an inclusive scan kernel

pub fn exclusive() -> Self

Create an exclusive scan kernel

pub fn execute( &self, _device: &GpuDevice, input: &[u32], output: &mut [u32], ) -> Result<()>

Execute the scan (prefix sum) operation (CPU fallback).

• Inclusive: output[i] = input[0] + … + input[i]
• Exclusive: output[0] = 0, output[i] = input[0] + … + input[i-1]

Wrapping arithmetic is used to avoid panics on overflow. output must have the same length as input.

Arguments

• _device - GPU device (CPU fallback: unused)
• input - Input data
• output - Output buffer for scan results

Errors

Returns an error if output.len() != input.len().

pub fn is_inclusive(&self) -> bool

Check if this is an inclusive scan