Struct ReductionKernel 

pub struct ReductionKernel {
    pub operation: String,
    pub input_size: usize,
    pub output_size: usize,
    pub num_axes: usize,
}

Reduction operations async kernel

Fields

operation: String

input_size: usize

output_size: usize

num_axes: usize

Implementations

impl ReductionKernel

pub fn new_sum(input_size: usize, output_size: usize, num_axes: usize) -> Self

pub fn new_mean(input_size: usize, output_size: usize, num_axes: usize) -> Self

pub fn new_max(input_size: usize, output_size: usize, num_axes: usize) -> Self

Trait Implementations

impl AsyncKernel for ReductionKernel

fn execute(&self) -> Pin<Box<dyn Future<Output = Result<()>> + Send>>

Execute the kernel asynchronously

fn name(&self) -> &str

Get kernel name for debugging

fn estimated_time_us(&self) -> u64

Get estimated execution time in microseconds

fn can_batch_with(&self, other: &dyn AsyncKernel) -> bool

Check if this kernel can be batched with another

fn priority(&self) -> Priority

Get kernel priority

fn supports_concurrency(&self) -> bool

Check if kernel supports concurrent execution

fn memory_requirements(&self) -> usize

Get memory requirements in bytes