---
id: pr-sglang-24271
repo: sgl-project/sglang
pr: 24271
title: '[KDA] Optimize prefill kernels with diagonal and recompute fuse'
author: yuan-luo
date: '2026-05-09'
url: https://github.com/sgl-project/sglang/pull/24271
source_category: upstream-code
architectures:
- sm90
tags:
- attention kernel
- attention_kv
- benchmark kernel
- hopper
- triton kernel
techniques: []
hardware_features:
- hopper
kernel_types:
- attention
languages:
- cuda-cpp
- triton
captured_at: '2026-05-16'
status: merged
merge_sha: a61a14f416c4003809a82112b4a591eec38a0a10
inclusion_reason: Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.
changed_paths:
- python/sglang/srt/layers/attention/fla/chunk_delta_h.py
- python/sglang/srt/layers/attention/fla/chunk_intra.py
- python/sglang/srt/layers/attention/fla/cumsum.py
- python/sglang/srt/layers/attention/fla/kda.py
artifact_dir: evidence/pull-bundles/sglang/gh-24271
head_ref_oid: a2459f3a8365827b08a40dc90a3cca9b2c4d558d
---
# [KDA] Optimize prefill kernels with diagonal and recompute fuse

- Repository: `sgl-project/sglang`
- PR: [24271](https://github.com/sgl-project/sglang/pull/24271)
- Decision: `Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.`
- Reason: Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.

## Changed Paths

- `python/sglang/srt/layers/attention/fla/chunk_delta_h.py`
- `python/sglang/srt/layers/attention/fla/chunk_intra.py`
- `python/sglang/srt/layers/attention/fla/cumsum.py`
- `python/sglang/srt/layers/attention/fla/kda.py`

## Evidence Bundle

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