Skip to main content

ReduceToOp

Enum ReduceToOp 

Source
#[non_exhaustive]
#[repr(u16)]
pub enum ReduceToOp { Sum = 0, Max = 2, Min = 3, Prod = 4, }
Expand description

Broadcast-reverse reduction op discriminant — ReduceToPlan.

The autograd primitive that undoes a forward BroadcastTo: for each output cell, reduce every input cell that broadcasts TO it. Distinct from ReduceKind because the reduction collapses an arbitrary set of axes in one launch (every dim where output_shape[d] == 1 while input_shape[d] != 1) rather than a single reduce_axis, so dispatch goes through its own plan shape (ReduceToPlan<T, N> in baracuda-kernels).

Discriminants mirror ReduceKind’s values for the same logical op so KernelSku::op tags consistently across the reduction family (hence the gap at 1 — there is no broadcast-reverse Mean).

Variants (Non-exhaustive)§

This enum is marked as non-exhaustive
Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.
§

Sum = 0

Sum over the broadcast set. Identity on an empty reduce set (any reduced input_shape[d] == 0): 0.

§

Max = 2

Maximum over the broadcast set. Identity on an empty reduce set: -FLT_MAX / -DBL_MAX per the kernel’s AccMax policy — the most-negative finite value for f32 / f64 outputs. For f16 / bf16 the f32 identity overflows the storage dtype on the final narrowing store and lands as -inf.

§

Min = 3

Minimum over the broadcast set. Identity on an empty reduce set: +FLT_MAX / +DBL_MAX per the kernel’s AccMin policy — the most-positive finite value for f32 / f64 outputs; +inf for f16 / bf16 (same narrowing overflow as Self::Max).

§

Prod = 4

Product over the broadcast set. Identity on an empty reduce set: 1.

Trait Implementations§

Source§

impl Clone for ReduceToOp

Source§

fn clone(&self) -> ReduceToOp

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

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

Performs copy-assignment from source. Read more
Source§

impl Copy for ReduceToOp

Source§

impl Debug for ReduceToOp

Source§

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

Formats the value using the given formatter. Read more
Source§

impl Eq for ReduceToOp

Source§

impl Hash for ReduceToOp

Source§

fn hash<__H>(&self, state: &mut __H)
where __H: Hasher,

Feeds this value into the given Hasher. Read more
1.3.0 · Source§

fn hash_slice<H>(data: &[Self], state: &mut H)
where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher. Read more
Source§

impl PartialEq for ReduceToOp

Source§

fn eq(&self, other: &ReduceToOp) -> bool

Tests for self and other values to be equal, and is used by ==.
1.0.0 (const: unstable) · Source§

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.
Source§

impl StructuralPartialEq for ReduceToOp

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.