---
id: pr-tensorrt-llm-12519
repo: NVIDIA/TensorRT-LLM
pr: 12519
title: '[#12634][feat] AutoDeploy: Support rank 256 MLA in flashinfer_mla'
author: bmarimuthu-nv
date: '2026-04-11'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/12519
source_category: upstream-code
architectures:
- sm100
tags:
- MLA
- attention_kv
techniques:
- cache-policy
hardware_features:
- blackwell
kernel_types:
- attention
- mla
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: 8a56f4b0de7a0b1522f5aa68e0de937cf67b5dc8
inclusion_reason: Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.
changed_paths:
- tensorrt_llm/_torch/auto_deploy/compile/piecewise_utils.py
- tensorrt_llm/_torch/auto_deploy/custom_ops/README.md
- tensorrt_llm/_torch/auto_deploy/custom_ops/mla/__init__.py
- tensorrt_llm/_torch/auto_deploy/custom_ops/mla/flashinfer_trtllm_mla.py
- tensorrt_llm/_torch/auto_deploy/transform/library/kvcache.py
- tests/unittest/auto_deploy/singlegpu/custom_ops/mla/test_flashinfer_trtllm_mla_op.py
- tests/unittest/auto_deploy/singlegpu/transformations/library/test_kv_cache.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-12519
head_ref_oid: 131e7b0bffd253a49cd8749a87f96235185a111a
---
# [#12634][feat] AutoDeploy: Support rank 256 MLA in flashinfer_mla

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [12519](https://github.com/NVIDIA/TensorRT-LLM/pull/12519)
- 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

- `tensorrt_llm/_torch/auto_deploy/compile/piecewise_utils.py`
- `tensorrt_llm/_torch/auto_deploy/custom_ops/README.md`
- `tensorrt_llm/_torch/auto_deploy/custom_ops/mla/__init__.py`
- `tensorrt_llm/_torch/auto_deploy/custom_ops/mla/flashinfer_trtllm_mla.py`
- `tensorrt_llm/_torch/auto_deploy/transform/library/kvcache.py`
- `tests/unittest/auto_deploy/singlegpu/custom_ops/mla/test_flashinfer_trtllm_mla_op.py`
- `tests/unittest/auto_deploy/singlegpu/transformations/library/test_kv_cache.py`

## Evidence Bundle

`evidence/pull-bundles/tensorrt-llm/gh-12519`
