From 4076dc49cbaa6346a9e01975582bf3df724ec368 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 5 Nov 2024 00:08:16 -0800 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 44a7cda12f..a8173d2453 100644 --- a/unsloth/kernels/rope_embedding.py +++ b/unsloth/kernels/rope_embedding.py @@ -25,9 +25,9 @@ def _rope_embedding( cos, cos_row_stride, sin, sin_row_stride, seqlen, - head_dim : tl.constexpr, - n_heads : tl.constexpr, - BACKWARD_PASS : tl.constexpr, + head_dim, + n_heads, + BACKWARD_PASS, BLOCK_SIZE : tl.constexpr, ): """