---
id: pr-vllm-21691
repo: vllm-project/vllm
pr: 21691
title: '[BugFix] Fix IMA FlashMLA full cuda-graph and DP + Update FlashMLA'
author: LucasWilkinson
date: '2025-08-08'
url: https://github.com/vllm-project/vllm/pull/21691
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- mla
techniques: []
hardware_features: []
kernel_types:
- attention
- mla
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: cd9b9de1fb009cf607403ba08961f2a3f869931d
inclusion_reason: kernel file changes
changed_paths:
- cmake/external_projects/flashmla.cmake
- vllm/attention/ops/flashmla.py
- vllm/v1/attention/backends/mla/flashmla.py
head_ref_oid: 98d2e860785fc92bd11cf353d9fb6c3ffa8b4eef
artifact_dir: evidence/pull-bundles/vllm/gh-21691
---
# [BugFix] Fix IMA FlashMLA full cuda-graph and DP + Update FlashMLA

- Repository: `vllm-project/vllm`
- PR: [21691](https://github.com/vllm-project/vllm/pull/21691)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `cmake/external_projects/flashmla.cmake`
- `vllm/attention/ops/flashmla.py`
- `vllm/v1/attention/backends/mla/flashmla.py`

## Evidence Bundle

`evidence/pull-bundles/vllm/gh-21691`
