---
id: pr-flashinfer-2645
repo: flashinfer-ai/flashinfer
pr: 2645
title: int16 Block-Scaled State and Stochastic Rounding for SSU (mamba)
author: ishovkun
date: '2026-03-04'
url: https://github.com/flashinfer-ai/flashinfer/pull/2645
source_category: upstream-code
architectures:
- sm100
tags:
- block-scale
techniques: []
hardware_features:
- block-scale
kernel_types: []
languages:
- cuda-cpp
- triton
captured_at: '2026-04-27'
status: merged
merge_sha: b0e7eb7744e5dd99ba7e5e17a31bf1797365f1ec
inclusion_reason: kernel file changes
changed_paths:
- .gitignore
- csrc/flashinfer_mamba_binding.cu
- csrc/selective_state_update.cu
- csrc/selective_state_update_customize_config.jinja
- csrc/selective_state_update_kernel_inst.cu
- flashinfer/aot.py
- flashinfer/jit/mamba/selective_state_update.py
- flashinfer/mamba/selective_state_update.py
- include/flashinfer/mamba/common.cuh
- include/flashinfer/mamba/conversion.cuh
- include/flashinfer/mamba/create_tensor_map.cuh
- include/flashinfer/mamba/kernel_selective_state_update_mtp.cuh
- include/flashinfer/mamba/kernel_selective_state_update_stp.cuh
- include/flashinfer/mamba/selective_state_update.cuh
- include/flashinfer/utils.cuh
- tests/mamba/test_philox_rounding.py
- tests/mamba/test_selective_state_update_mtp.py
- tests/mamba/test_selective_state_update_stp.py
- tests/mamba/triton_reference/selective_state_update.py
- tests/mamba/utils.py
head_ref_oid: 10a36c47ebe6d8b812e446fc25f961e8be411062
artifact_dir: evidence/pull-bundles/flashinfer/gh-2645
---
# int16 Block-Scaled State and Stochastic Rounding for SSU (mamba)

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

## Changed Paths

- `.gitignore`
- `csrc/flashinfer_mamba_binding.cu`
- `csrc/selective_state_update.cu`
- `csrc/selective_state_update_customize_config.jinja`
- `csrc/selective_state_update_kernel_inst.cu`
- `flashinfer/aot.py`
- `flashinfer/jit/mamba/selective_state_update.py`
- `flashinfer/mamba/selective_state_update.py`
- `include/flashinfer/mamba/common.cuh`
- `include/flashinfer/mamba/conversion.cuh`
- `include/flashinfer/mamba/create_tensor_map.cuh`
- `include/flashinfer/mamba/kernel_selective_state_update_mtp.cuh`
- `include/flashinfer/mamba/kernel_selective_state_update_stp.cuh`
- `include/flashinfer/mamba/selective_state_update.cuh`
- `include/flashinfer/utils.cuh`
- `tests/mamba/test_philox_rounding.py`
- `tests/mamba/test_selective_state_update_mtp.py`
- `tests/mamba/test_selective_state_update_stp.py`
- `tests/mamba/triton_reference/selective_state_update.py`
- `tests/mamba/utils.py`

## Evidence Bundle

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