---
id: pr-vllm-12777
repo: vllm-project/vllm
pr: 12777
title: '[Kernel] Make rotary_embedding ops more flexible with input shape'
author: Isotr0py
date: '2025-02-06'
url: https://github.com/vllm-project/vllm/pull/12777
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: 85ac82d228ef6af4e8fc6332d918133e783a0fdb
inclusion_reason: kernel file changes
changed_paths:
- csrc/pos_encoding_kernels.cu
- tests/kernels/test_pos_encoding.py
- vllm/attention/backends/mla/utils.py
- vllm/model_executor/models/deepseek_v2.py
head_ref_oid: dd2d588ca02afe0b1ce7085f6dd1d0331c6ef306
artifact_dir: evidence/pull-bundles/vllm/gh-12777
---
# [Kernel] Make rotary_embedding ops more flexible with input shape

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

## Changed Paths

- `csrc/pos_encoding_kernels.cu`
- `tests/kernels/test_pos_encoding.py`
- `vllm/attention/backends/mla/utils.py`
- `vllm/model_executor/models/deepseek_v2.py`

## Evidence Bundle

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