---
id: pr-sglang-21511
repo: sgl-project/sglang
pr: 21511
title: '[AMD] Enable FP8 KV cache and FP8 attention kernel for NSA on MI300/MI355 with TileLang backend'
author: 1am9trash
date: '2026-04-03'
url: https://github.com/sgl-project/sglang/pull/21511
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- fp8
- mla
techniques: []
hardware_features:
- fp8
kernel_types:
- attention
- mla
languages:
- python
- tilelang
captured_at: '2026-04-17'
status: merged
merge_sha: 7431db7392fe5b4a5beb8c296a6dcba2b4bb9ba0
inclusion_reason: kernel file changes
changed_paths:
- docker/rocm.Dockerfile
- python/sglang/srt/layers/attention/nsa/tilelang_kernel.py
- python/sglang/srt/mem_cache/memory_pool.py
- python/sglang/srt/mem_cache/utils.py
- python/sglang/srt/model_executor/model_runner_kv_cache_mixin.py
- python/sglang/srt/models/deepseek_common/attention_forward_methods/forward_mla.py
head_ref_oid: f84a830f9da398661c64a958552e7465357d5a55
artifact_dir: evidence/pull-bundles/sglang/gh-21511
---
# [AMD] Enable FP8 KV cache and FP8 attention kernel for NSA on MI300/MI355 with TileLang backend

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

## Changed Paths

- `docker/rocm.Dockerfile`
- `python/sglang/srt/layers/attention/nsa/tilelang_kernel.py`
- `python/sglang/srt/mem_cache/memory_pool.py`
- `python/sglang/srt/mem_cache/utils.py`
- `python/sglang/srt/model_executor/model_runner_kv_cache_mixin.py`
- `python/sglang/srt/models/deepseek_common/attention_forward_methods/forward_mla.py`

## Evidence Bundle

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