From a5f26f4f762415e73a33959d8796300de45a0458 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 23 Jul 2025 03:26:16 -0700
Subject: [PATCH] Update rope_embedding.py

---
 unsloth/kernels/rope_embedding.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/unsloth/kernels/rope_embedding.py b/unsloth/kernels/rope_embedding.py
index 1c981b3fc9..a06d81a512 100644
--- a/unsloth/kernels/rope_embedding.py
+++ b/unsloth/kernels/rope_embedding.py
@@ -19,9 +19,9 @@ from .utils import calculate_settings, torch_gpu_device
 ROPE_GROUP_SIZE : int = 4
 
 def _rope_embedding(
-    Q,     Q_row_stride,
-    cos, cos_row_stride,
-    sin, sin_row_stride,
+    Q,     Q_row_stride: tl.constexpr,
+    cos, cos_row_stride: tl.constexpr,
+    sin, sin_row_stride: tl.constexpr,
     seqlen,
     head_dim      : tl.constexpr,
     n_heads       : tl.constexpr,