---
id: pr-sglang-11812
repo: sgl-project/sglang
pr: 11812
title: Support piecewise cuda graph for MLA
author: ispobock
date: '2025-11-10'
url: https://github.com/sgl-project/sglang/pull/11812
source_category: upstream-code
architectures:
- sm100
tags: []
techniques: []
hardware_features: []
kernel_types: []
languages: []
captured_at: '2026-05-16'
status: merged
merge_sha: db24d34603d3797ece082f0b8f7e73e16c65f22a
inclusion_reason: defer
changed_paths:
- python/sglang/srt/compilation/compile.py
- python/sglang/srt/compilation/piecewise_context_manager.py
- python/sglang/srt/layers/attention/flashinfer_mla_backend.py
- python/sglang/srt/layers/radix_attention.py
- python/sglang/srt/layers/rotary_embedding.py
- python/sglang/srt/model_executor/model_runner.py
- python/sglang/srt/model_executor/piecewise_cuda_graph_runner.py
- python/sglang/srt/models/deepseek_v2.py
- test/srt/run_suite.py
- test/srt/test_piecewise_cuda_graph.py
artifact_dir: evidence/pull-bundles/sglang/gh-11812
head_ref_oid: 089b0e354030635bbf920a6f79b552ef91fb510c
---
# Support piecewise cuda graph for MLA

- Repository: `sgl-project/sglang`
- PR: [11812](https://github.com/sgl-project/sglang/pull/11812)
- Decision: `defer`
- Reason: defer

## Changed Paths

- `python/sglang/srt/compilation/compile.py`
- `python/sglang/srt/compilation/piecewise_context_manager.py`
- `python/sglang/srt/layers/attention/flashinfer_mla_backend.py`
- `python/sglang/srt/layers/radix_attention.py`
- `python/sglang/srt/layers/rotary_embedding.py`
- `python/sglang/srt/model_executor/model_runner.py`
- `python/sglang/srt/model_executor/piecewise_cuda_graph_runner.py`
- `python/sglang/srt/models/deepseek_v2.py`
- `test/srt/run_suite.py`
- `test/srt/test_piecewise_cuda_graph.py`

## Evidence Bundle

`evidence/pull-bundles/sglang/gh-11812`
