ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update rope_embedding.py

Browse source
This commit is contained in:
Daniel Han 2025-07-23 03:26:16 -07:00
commit a5f26f4f76
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/kernels/rope_embedding.py
View file

@ -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,