Skip to content

Add state-passing context parallelism for Mamba2 - #6808

Open
shcho1118 wants to merge 5 commits into
NVIDIA:devfrom
shcho1118:feature/mamba2-state-passing-cp-upstream
Open

Add state-passing context parallelism for Mamba2#6808
shcho1118 wants to merge 5 commits into
NVIDIA:devfrom
shcho1118:feature/mamba2-state-passing-cp-upstream

Add a benchmark for the Mamba2 context-parallel paths

622a768
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 24, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO