---
id: pr-tensorrt-llm-13677
repo: NVIDIA/TensorRT-LLM
pr: 13677
title: '[#11823][feat] AutoDeploy''s mla chunked prefill loop support'
author: MrGeva
date: '2026-05-05'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/13677
source_category: upstream-code
architectures:
- sm100
tags:
- attention kernel
- cuda kernel
techniques: []
hardware_features: []
kernel_types:
- attention
- mla
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: a90389e565f4c29c74131904cc55e87d28a83ac0
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- tensorrt_llm/_torch/auto_deploy/custom_ops/mla/trtllm_mla.py
- tests/unittest/auto_deploy/singlegpu/custom_ops/mla/test_trtllm_mla_op.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-13677
head_ref_oid: d6c018dd5d9ab18a28a4f7620adedb2b66405041
---
# [#11823][feat] AutoDeploy's mla chunked prefill loop support

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [13677](https://github.com/NVIDIA/TensorRT-LLM/pull/13677)
- Decision: `Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.`
- Reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.

## Changed Paths

- `tensorrt_llm/_torch/auto_deploy/custom_ops/mla/trtllm_mla.py`
- `tests/unittest/auto_deploy/singlegpu/custom_ops/mla/test_trtllm_mla_op.py`

## Evidence Bundle

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