fix rotary_emb

Author: none

lightllm/models/deepseek2/triton_kernel/rotary_emb.py

@@ -21,8 +21,8 @@ def _rotary_kernel(
     stride_sinbs,
     stride_sind,
     max_total_len,
-    HEAD_Q,
-    HEAD_K,
+    HEAD_Q: tl.constexpr,
+    HEAD_K: tl.constexpr,
     BLOCK_SEQ: tl.constexpr,
     BLOCK_DMODEL: tl.constexpr,
     NUM_STAGE: tl.constexpr,