---
id: pr-vllm-12676
repo: vllm-project/vllm
pr: 12676
title: '[Perf] Mem align KV caches for CUDA devices (MLA perf improvement)'
author: LucasWilkinson
date: '2025-02-05'
url: https://github.com/vllm-project/vllm/pull/12676
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- decode
- mla
techniques: []
hardware_features: []
kernel_types:
- attention
- decode
- mla
languages:
- cuda-cpp
- triton
captured_at: '2026-04-27'
status: merged
merge_sha: 75e94309e8d8919e0daea041f6cd81a4b8c09060
inclusion_reason: kernel file changes
changed_paths:
- csrc/cache.h
- csrc/cache_kernels.cu
- csrc/torch_bindings.cpp
- tests/kernels/test_cache.py
- vllm/_custom_ops.py
- vllm/attention/backends/triton_mla.py
- vllm/attention/ops/triton_decode_attention.py
- vllm/envs.py
- vllm/utils.py
- vllm/worker/cache_engine.py
head_ref_oid: 93e43847270cef6f0bef4040f7b121e871209587
artifact_dir: evidence/pull-bundles/vllm/gh-12676
---
# [Perf] Mem align KV caches for CUDA devices (MLA perf improvement)

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

## Changed Paths

- `csrc/cache.h`
- `csrc/cache_kernels.cu`
- `csrc/torch_bindings.cpp`
- `tests/kernels/test_cache.py`
- `vllm/_custom_ops.py`
- `vllm/attention/backends/triton_mla.py`
- `vllm/attention/ops/triton_decode_attention.py`
- `vllm/envs.py`
- `vllm/utils.py`
- `vllm/worker/cache_engine.py`

## Evidence Bundle

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