Expand description
CPU forward-pass primitives: RMS-norm, SiLU, sparse SwiGLU FFN.
Functionsยง
- rms_
norm - RMS normalization with explicit weight semantics.
- silu
- SiLU activation function.
- sparse_
ffn_ forward - Sparse SwiGLU FFN: compute only
active_indicesneurons. - sparse_
ffn_ forward_ pair - Fused two-position sparse FFN: gate/up/down weight rows are streamed from memory ONCE for both positions. Bit-identical to two single calls (per-position accumulation order is unchanged).