---
id: pr-flashinfer-804
repo: flashinfer-ai/flashinfer
pr: 804
title: 'perf: memory efficient deepseek mla fused page-attention kernel'
author: yzh119
date: '2025-02-12'
url: https://github.com/flashinfer-ai/flashinfer/pull/804
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- kernel-fusion
- mla
- prefill
techniques:
- kernel-fusion
hardware_features: []
kernel_types:
- attention
- mla
- prefill
languages:
- cuda-cpp
captured_at: '2026-04-17'
status: merged
merge_sha: 106e6fc24a80ee2aa534dfd29274bb67091dce65
inclusion_reason: kernel file changes
changed_paths:
- benchmarks/bench_deepseek_mla.py
- csrc/batch_mla_config.jinja
- csrc/batch_mla_plan.cu
- csrc/batch_mla_pybind.cu
- csrc/batch_mla_run.cu
- flashinfer/__init__.py
- flashinfer/jit/__init__.py
- flashinfer/jit/attention.py
- flashinfer/mla.py
- include/flashinfer/attention/heap.h
- include/flashinfer/attention/mla_fa2.cuh
- include/flashinfer/attention/mla_params.cuh
- include/flashinfer/attention/prefill.cuh
- include/flashinfer/attention/scheduler.cuh
- include/flashinfer/attention/variant_helper.cuh
- tests/test_deepseek_mla.py
- tests/test_deepseek_prefill.py
head_ref_oid: 0aaf837d142ad7cc85e05abac2d32695dba8f854
artifact_dir: evidence/pull-bundles/flashinfer/gh-804
---
# perf: memory efficient deepseek mla fused page-attention kernel

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

## Changed Paths

- `benchmarks/bench_deepseek_mla.py`
- `csrc/batch_mla_config.jinja`
- `csrc/batch_mla_plan.cu`
- `csrc/batch_mla_pybind.cu`
- `csrc/batch_mla_run.cu`
- `flashinfer/__init__.py`
- `flashinfer/jit/__init__.py`
- `flashinfer/jit/attention.py`
- `flashinfer/mla.py`
- `include/flashinfer/attention/heap.h`
- `include/flashinfer/attention/mla_fa2.cuh`
- `include/flashinfer/attention/mla_params.cuh`
- `include/flashinfer/attention/prefill.cuh`
- `include/flashinfer/attention/scheduler.cuh`
- `include/flashinfer/attention/variant_helper.cuh`
- `tests/test_deepseek_mla.py`
- `tests/test_deepseek_prefill.py`

## Evidence Bundle

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