Struct ReducePlan 

pub struct ReducePlan<T: Element, const N: usize> { /* private fields */ }

Single-axis reduction plan — see module docs for the wired matrix, workspace, and precision guarantees.

T: Element is the element type (f32 / f64 / f16 / bf16). const N: usize is the tensor rank.

Implementations

impl<T: Element, const N: usize> ReducePlan<T, N>

pub fn select( _stream: &Stream, desc: &ReduceDescriptor<N>, _pref: PlanPreference, ) -> Result<Self>

Pick a kernel for desc.

pub fn can_implement(&self, args: &ReduceArgs<'_, T, N>) -> Result<()>

Validate args.

pub fn workspace_size(&self) -> usize

Workspace size in bytes. Always 0 for the naive trailblazer.

pub fn sku(&self) -> KernelSku

Identity of the kernel this plan picked.

pub fn precision_guarantee(&self) -> PrecisionGuarantee

Numerical guarantees for this plan’s kernel.

pub fn run( &self, stream: &Stream, _workspace: Workspace<'_>, args: ReduceArgs<'_, T, N>, ) -> Result<()>

Launch.