---
id: pr-flashinfer-3132
repo: flashinfer-ai/flashinfer
pr: 3132
title: '[CuTe DSL] Fix FP8 MLA persistent perf regression and ProxyKind cu13 wheel breakage'
author: pgera
date: '2026-04-22'
url: https://github.com/flashinfer-ai/flashinfer/pull/3132
source_category: upstream-code
architectures:
- sm100
tags:
- attention_kv
- blackwell
- compiler_runtime
- cute dsl
- cutlass
- fp8
- mla
techniques:
- async-copy
- persistent-kernels
hardware_features:
- blackwell
- fp8
- tma
kernel_types:
- attention
- compiler
- decode
- mla
- softmax
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: f00ce3603e9e6ec4460f3fed7a8a5a0eabbc2169
inclusion_reason: Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.
changed_paths:
- flashinfer/cute_dsl/attention/mla_decode.py
- flashinfer/cute_dsl/attention/mla_decode_fp8.py
- flashinfer/cute_dsl/attention/prefill.py
- flashinfer/cute_dsl/attention/roles/correction.py
- flashinfer/cute_dsl/attention/roles/mla_mma.py
- flashinfer/cute_dsl/attention/roles/mla_mma_fp8.py
- flashinfer/cute_dsl/attention/roles/mma.py
- flashinfer/cute_dsl/attention/roles/softmax.py
artifact_dir: evidence/pull-bundles/flashinfer/gh-3132
head_ref_oid: 81163c5d70e044ba96a106eca696c4012caa1ddf
---
# [CuTe DSL] Fix FP8 MLA persistent perf regression and ProxyKind cu13 wheel breakage

- Repository: `flashinfer-ai/flashinfer`
- PR: [3132](https://github.com/flashinfer-ai/flashinfer/pull/3132)
- 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

- `flashinfer/cute_dsl/attention/mla_decode.py`
- `flashinfer/cute_dsl/attention/mla_decode_fp8.py`
- `flashinfer/cute_dsl/attention/prefill.py`
- `flashinfer/cute_dsl/attention/roles/correction.py`
- `flashinfer/cute_dsl/attention/roles/mla_mma.py`
- `flashinfer/cute_dsl/attention/roles/mla_mma_fp8.py`
- `flashinfer/cute_dsl/attention/roles/mma.py`
- `flashinfer/cute_dsl/attention/roles/softmax.py`

## Evidence Bundle

`evidence/pull-bundles/flashinfer/gh-3132`
