Skip to main content

recognize_reduction

Function recognize_reduction 

Source
pub fn recognize_reduction(
    stmts: &[Stmt<'_>],
    _interner: &Interner,
) -> Result<ReductionSpec, ReductionReject>
Expand description

Recognize the rmsnorm / softmax normalize-loop idiom the sugar desugars to — a counted loop that Pushes X[p] / denom (rmsnorm) or e^(X[p]-m) / s (softmax) into Y — anywhere (through Splice).