From 99bd67c9b29851f758c7d22caa8fc57fd5af3e4f Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Thu, 26 Feb 2026 18:08:54 +0200
Subject: [PATCH] kv-cache : fix can_shift() check to take into account M-RoPE
 (#19928)

---
 src/llama-kv-cache.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/llama-kv-cache.cpp b/src/llama-kv-cache.cpp
index cb702b2a59..6b668ee9ab 100644
--- a/src/llama-kv-cache.cpp
+++ b/src/llama-kv-cache.cpp
@@ -978,6 +978,9 @@ bool llama_kv_cache::get_can_shift() const {
     if (model.arch == LLM_ARCH_STEP35) {
         return false;
     }
+    if (hparams.n_pos_per_embd() > 1) {
+        return false;
+    }
     return true;
 }