---
id: pr-vllm-12528
repo: vllm-project/vllm
pr: 12528
title: '[Attention] MLA decode optimizations'
author: LucasWilkinson
date: '2025-01-31'
url: https://github.com/vllm-project/vllm/pull/12528
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: cabaf4eff3c7df30d785769d5a0a1fa1a1c48a8a
inclusion_reason: kernel file changes
changed_paths:
- csrc/cache.h
- csrc/cache_kernels.cu
- csrc/torch_bindings.cpp
- tests/kernels/test_triton_decode_attention.py
- tests/plugins/vllm_add_dummy_platform/vllm_add_dummy_platform/dummy_platform.py
- tests/weight_loading/models.txt
- tests/weight_loading/run_model_weight_loading_test.sh
- vllm/_custom_ops.py
- vllm/attention/backends/abstract.py
- vllm/attention/backends/mla/__init__.py
- vllm/attention/backends/mla/utils.py
- vllm/attention/backends/triton_mla.py
- vllm/attention/backends/utils.py
- vllm/attention/layer.py
- vllm/attention/ops/triton_decode_attention.py
- vllm/attention/selector.py
- vllm/config.py
- vllm/engine/arg_utils.py
- vllm/envs.py
- vllm/model_executor/model_loader/loader.py
- vllm/model_executor/models/deepseek_v2.py
- vllm/platforms/cpu.py
- vllm/platforms/cuda.py
- vllm/platforms/hpu.py
- vllm/platforms/interface.py
- vllm/platforms/openvino.py
- vllm/platforms/rocm.py
- vllm/platforms/tpu.py
- vllm/platforms/xpu.py
- vllm/worker/cache_engine.py
- vllm/worker/model_runner.py
head_ref_oid: becc99c5ab5fa23807846b7f72ba24df69c17ec0
artifact_dir: evidence/pull-bundles/vllm/gh-12528
---
# [Attention] MLA decode optimizations

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

## Changed Paths

- `csrc/cache.h`
- `csrc/cache_kernels.cu`
- `csrc/torch_bindings.cpp`
- `tests/kernels/test_triton_decode_attention.py`
- `tests/plugins/vllm_add_dummy_platform/vllm_add_dummy_platform/dummy_platform.py`
- `tests/weight_loading/models.txt`
- `tests/weight_loading/run_model_weight_loading_test.sh`
- `vllm/_custom_ops.py`
- `vllm/attention/backends/abstract.py`
- `vllm/attention/backends/mla/__init__.py`
- `vllm/attention/backends/mla/utils.py`
- `vllm/attention/backends/triton_mla.py`
- `vllm/attention/backends/utils.py`
- `vllm/attention/layer.py`
- `vllm/attention/ops/triton_decode_attention.py`
- `vllm/attention/selector.py`
- `vllm/config.py`
- `vllm/engine/arg_utils.py`
- `vllm/envs.py`
- `vllm/model_executor/model_loader/loader.py`
- `vllm/model_executor/models/deepseek_v2.py`
- `vllm/platforms/cpu.py`
- `vllm/platforms/cuda.py`
- `vllm/platforms/hpu.py`
- `vllm/platforms/interface.py`
- `vllm/platforms/openvino.py`
- `vllm/platforms/rocm.py`
- `vllm/platforms/tpu.py`
- `vllm/platforms/xpu.py`
- `vllm/worker/cache_engine.py`
- `vllm/worker/model_runner.py`

## Evidence Bundle

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