---
id: pr-sglang-20409
repo: sgl-project/sglang
pr: 20409
title: '[AMD][AITER] Guard _use_mla_ps_kernel with self.use_mla in draft_extend_v2 paths'
author: ChuanLi1101
date: '2026-03-18'
url: https://github.com/sgl-project/sglang/pull/20409
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- mla
techniques: []
hardware_features: []
kernel_types:
- attention
- mla
languages:
- python
captured_at: '2026-04-17'
status: merged
merge_sha: 93422f27d6f41c2cfb0315c9569c98fc3a55a322
inclusion_reason: kernel file changes
changed_paths:
- python/sglang/srt/layers/attention/aiter_backend.py
head_ref_oid: 42cdc097cd53acc4898105b71029ac1603dfa1dd
artifact_dir: evidence/pull-bundles/sglang/gh-20409
---
# [AMD][AITER] Guard _use_mla_ps_kernel with self.use_mla in draft_extend_v2 paths

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

## Changed Paths

- `python/sglang/srt/layers/attention/aiter_backend.py`

## Evidence Bundle

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