---
id: pr-sglang-11664
repo: sgl-project/sglang
pr: 11664
title: Use trtllm_mla decode kernel for draft extend in speculative decoding
author: Qiaolin-Yu
date: '2025-10-21'
url: https://github.com/sgl-project/sglang/pull/11664
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- decode
- mla
techniques: []
hardware_features: []
kernel_types:
- attention
- decode
- mla
languages:
- python
captured_at: '2026-04-27'
status: merged
merge_sha: d9a20fd28ae67c0b7666ee40dc86f73239a95c3d
inclusion_reason: kernel file changes
changed_paths:
- python/sglang/srt/layers/attention/trtllm_mla_backend.py
- python/sglang/test/attention/test_trtllm_mla_backend.py
head_ref_oid: 4584857736022fd5f1ab6464dda49d8a413b0fe5
artifact_dir: evidence/pull-bundles/sglang/gh-11664
---
# Use trtllm_mla decode kernel for draft extend in speculative decoding

- Repository: `sgl-project/sglang`
- PR: [11664](https://github.com/sgl-project/sglang/pull/11664)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `python/sglang/srt/layers/attention/trtllm_mla_backend.py`
- `python/sglang/test/attention/test_trtllm_mla_backend.py`

## Evidence Bundle

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