---
id: pr-flashinfer-2237
repo: flashinfer-ai/flashinfer
pr: 2237
title: '[feat] Integrate SGLang concat_mla_k kernel into flashinfer'
author: jiahanc
date: '2025-12-18'
url: https://github.com/flashinfer-ai/flashinfer/pull/2237
source_category: upstream-code
architectures:
- sm100
tags:
- mla
techniques: []
hardware_features: []
kernel_types:
- mla
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: de60eca55bf766e1c6f4240733f69360c4cf9b98
inclusion_reason: kernel file changes
changed_paths:
- benchmarks/bench_concat_mla.py
- csrc/concat_mla.cu
- flashinfer/concat_ops.py
- flashinfer/dsv3_ops/__init__.py
- flashinfer/jit/dsv3_optimizations.py
- include/flashinfer/concat_mla.cuh
- include/flashinfer/utils.cuh
head_ref_oid: d046645ada24fc20ca6b6b9e95f616c60f84dd63
artifact_dir: evidence/pull-bundles/flashinfer/gh-2237
---
# [feat] Integrate SGLang concat_mla_k kernel into flashinfer

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

## Changed Paths

- `benchmarks/bench_concat_mla.py`
- `csrc/concat_mla.cu`
- `flashinfer/concat_ops.py`
- `flashinfer/dsv3_ops/__init__.py`
- `flashinfer/jit/dsv3_optimizations.py`
- `include/flashinfer/concat_mla.cuh`
- `include/flashinfer/utils.cuh`

## Evidence Bundle

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