---
id: pr-flashinfer-3235
repo: flashinfer-ai/flashinfer
pr: 3235
title: Support Kimi K2.5 H64 CuTe DSL MLA decode
author: saltyminty
date: '2026-05-11'
url: https://github.com/flashinfer-ai/flashinfer/pull/3235
source_category: upstream-code
architectures:
- sm100
tags:
- attention_kv
- benchmark
- compiler_runtime
- cute dsl
- fp8
- mla
- scheduler_autotune
- sm100
techniques:
- autotuning
- persistent-kernels
- tile-scheduling
hardware_features:
- blackwell
- fp8
kernel_types:
- attention
- compiler
- decode
- mla
- scheduler
languages:
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: 2d0e0efacb011b4a365c7e4999ffe118f4d610e1
inclusion_reason: Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.
changed_paths:
- flashinfer/cute_dsl/attention/mla_config.py
- flashinfer/cute_dsl/attention/mla_decode.py
- flashinfer/cute_dsl/attention/mla_decode_fp8.py
- flashinfer/cute_dsl/attention/scheduler/mla_persistent.py
- flashinfer/cute_dsl/attention/wrappers/batch_mla.py
- tests/attention/test_cute_dsl_mla_decode.py
- tests/attention/test_trtllm_gen_mla.py
artifact_dir: evidence/pull-bundles/flashinfer/gh-3235
head_ref_oid: b4d193cde01671ad9b47e9c7a760b9e4d41b7a01
---
# Support Kimi K2.5 H64 CuTe DSL MLA decode

- Repository: `flashinfer-ai/flashinfer`
- PR: [3235](https://github.com/flashinfer-ai/flashinfer/pull/3235)
- 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_config.py`
- `flashinfer/cute_dsl/attention/mla_decode.py`
- `flashinfer/cute_dsl/attention/mla_decode_fp8.py`
- `flashinfer/cute_dsl/attention/scheduler/mla_persistent.py`
- `flashinfer/cute_dsl/attention/wrappers/batch_mla.py`
- `tests/attention/test_cute_dsl_mla_decode.py`
- `tests/attention/test_trtllm_gen_mla.py`

## Evidence Bundle

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