---
id: pr-flashinfer-2709
repo: flashinfer-ai/flashinfer
pr: 2709
title: Mamba2 SSD Combined Forward Pass (Blackwell CuTe DSL Kernel)
author: ishovkun
date: '2026-03-23'
url: https://github.com/flashinfer-ai/flashinfer/pull/2709
source_category: upstream-code
architectures:
- sm100
tags:
- tile-scheduling
techniques:
- tile-scheduling
hardware_features: []
kernel_types: []
languages:
- cuda-cpp
- triton
captured_at: '2026-04-27'
status: merged
merge_sha: c0dbc38c1f7df0a22c72213685f12393e826f6d6
inclusion_reason: kernel file changes
changed_paths:
- benchmarks/bench_mamba_ssd_combined.py
- csrc/seq_chunk_cumsum.cu
- csrc/seq_chunk_cumsum_jit_binding.cu
- flashinfer/jit/mamba/__init__.py
- flashinfer/jit/mamba/seq_chunk_cumsum.py
- flashinfer/mamba/__init__.py
- flashinfer/mamba/ssd_combined.py
- flashinfer/mamba/ssd_kernel.py
- flashinfer/mamba/ssd_tile_scheduler.py
- flashinfer/triton/kernels/ssd_chunk_state.py
- include/flashinfer/mamba/seq_chunk_cumsum.cuh
- tests/mamba/test_chunk_scan_combined.py
- tests/mamba/test_seq_chunk_cumsum.py
- tests/mamba/triton_reference/ssd_bmm.py
- tests/mamba/triton_reference/ssd_chunk_scan.py
- tests/mamba/triton_reference/ssd_chunk_state.py
- tests/mamba/triton_reference/ssd_combined.py
- tests/mamba/triton_reference/ssd_state_passing.py
head_ref_oid: d30316c355449276c5c0d6862a043e0466ea1aa0
artifact_dir: evidence/pull-bundles/flashinfer/gh-2709
---
# Mamba2 SSD Combined Forward Pass (Blackwell CuTe DSL Kernel)

- Repository: `flashinfer-ai/flashinfer`
- PR: [2709](https://github.com/flashinfer-ai/flashinfer/pull/2709)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `benchmarks/bench_mamba_ssd_combined.py`
- `csrc/seq_chunk_cumsum.cu`
- `csrc/seq_chunk_cumsum_jit_binding.cu`
- `flashinfer/jit/mamba/__init__.py`
- `flashinfer/jit/mamba/seq_chunk_cumsum.py`
- `flashinfer/mamba/__init__.py`
- `flashinfer/mamba/ssd_combined.py`
- `flashinfer/mamba/ssd_kernel.py`
- `flashinfer/mamba/ssd_tile_scheduler.py`
- `flashinfer/triton/kernels/ssd_chunk_state.py`
- `include/flashinfer/mamba/seq_chunk_cumsum.cuh`
- `tests/mamba/test_chunk_scan_combined.py`
- `tests/mamba/test_seq_chunk_cumsum.py`
- `tests/mamba/triton_reference/ssd_bmm.py`
- `tests/mamba/triton_reference/ssd_chunk_scan.py`
- `tests/mamba/triton_reference/ssd_chunk_state.py`
- `tests/mamba/triton_reference/ssd_combined.py`
- `tests/mamba/triton_reference/ssd_state_passing.py`

## Evidence Bundle

`evidence/pull-bundles/flashinfer/gh-2709`
