---
id: pr-flashinfer-2387
repo: flashinfer-ai/flashinfer
pr: 2387
title: A Blackwell-optimized version of selective_state_update (mamba)
author: ishovkun
date: '2026-01-22'
url: https://github.com/flashinfer-ai/flashinfer/pull/2387
source_category: upstream-code
architectures:
- sm100
tags:
- tcgen05
- decode
techniques:
- warp-specialization
- pipeline-stages
- double-buffering
hardware_features:
- tma
- tmem
- mbarrier
- tcgen05
kernel_types:
- fused-kernel
- decode
languages:
- cuda-cpp
- python
captured_at: 2026-04-16
status: merged
merge_sha: 18804cd51734cccf807356d017733bc757677f15
inclusion_reason: manually curated, Blackwell kernel relevance
changed_paths:
- flashinfer/aot.py
- flashinfer/jit/mamba/__init__.py
- flashinfer/jit/mamba/selective_state_update.py
- flashinfer/mamba/selective_state_update.py
- include/flashinfer/mamba/selective_state_update.cuh
- uv.lock
artifact_dir: evidence/pull-bundles/flashinfer/gh-2387
head_ref_oid: bfe3867d2c9ced629326dc6c96ec81e0b32482d0
---
# A Blackwell-optimized version of selective_state_update (mamba)

- Repository: `flashinfer-ai/flashinfer`
- PR: [2387](https://github.com/flashinfer-ai/flashinfer/pull/2387)
- Decision: `manually curated, Blackwell kernel relevance`
- Reason: manually curated, Blackwell kernel relevance

## Changed Paths

- `flashinfer/aot.py`
- `flashinfer/jit/mamba/__init__.py`
- `flashinfer/jit/mamba/selective_state_update.py`
- `flashinfer/mamba/selective_state_update.py`
- `include/flashinfer/mamba/selective_state_update.cuh`
- `uv.lock`

## Evidence Bundle

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