Add an opt-in Helion kernel backend for the Mamba-2 SSD path - #1012

Open

shcho1118 wants to merge 13 commits into

state-spaces:mainfrom

shcho1118:helion-kernels

Commits on Aug 3, 2026

helion: add shared runtime and autotune utilities
sanghun-cho
committed
e538519
helion: add SSD forward kernels behind an opt-in dispatch
sanghun-cho
committed
af5ddea
helion: add fused SSD input-gradient kernel
sanghun-cho
committed
88ee7a9
helion: add SSD state-propagation backward kernels
sanghun-cho
committed
067cd5f
helion: add SSD chunk-state dB backward kernel
sanghun-cho
committed
929b69b
helion: add SSD dC and stable dA backward kernels
sanghun-cho
committed
1bde619
helion: use Helion backward kernel for gated normalization
sanghun-cho
committed
e87dfe2
helion: add MambaMixer autotune harness
sanghun-cho
committed
0ba48b5
helion: add per-kernel comparison tests against Triton
sanghun-cho
committed
e09267b
helion: add per-stage backend benchmark
sanghun-cho
committed
a4c2cf5
helion: check in an example autotuned config set
sanghun-cho
committed
0cae04f
helion: document the backend

sanghun-cho
and
claude
committed
0ae828b

Commits on Aug 4, 2026

helion: harden MambaMixerMin init and add pass-path diff stats to test_kernels

sanghun-cho
and
claude
committed
396cbc3