---
id: pr-sglang-21213
repo: sgl-project/sglang
pr: 21213
title: '[AMD]: Support MLA with nhead<16 and FP8 KV cache for TP=8 (Kimi K2.5…'
author: ZiguanWang
date: '2026-04-05'
url: https://github.com/sgl-project/sglang/pull/21213
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- fp4
- fp8
- mla
techniques: []
hardware_features:
- fp4
- fp8
kernel_types:
- attention
- mla
languages:
- python
captured_at: '2026-04-17'
status: merged
merge_sha: dd49127fe612800d2f2aa258c9b7086043f103fa
inclusion_reason: kernel file changes
changed_paths:
- python/sglang/srt/layers/attention/aiter_backend.py
- test/registered/amd/accuracy/mi35x/test_kimi_k25_mxfp4_eval_mi35x.py
- test/registered/amd/test_kimi_k25_mxfp4.py
head_ref_oid: b646ebf7d0c67d6e19ceb68cffde2389eb229fa6
artifact_dir: evidence/pull-bundles/sglang/gh-21213
---
# [AMD]: Support MLA with nhead<16 and FP8 KV cache for TP=8 (Kimi K2.5…

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

## Changed Paths

- `python/sglang/srt/layers/attention/aiter_backend.py`
- `test/registered/amd/accuracy/mi35x/test_kimi_k25_mxfp4_eval_mi35x.py`
- `test/registered/amd/test_kimi_k25_mxfp4.py`

## Evidence Bundle

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