[Feat] upgrade pytorch2.6 (#5417)

2025-04-21 07:06:34 +08:00
parent 475e2e378a
commit 417b44eba8
7 changed files with 8 additions and 8 deletions
--- a/python/sglang/srt/layers/dp_attention.py
+++ b/python/sglang/srt/layers/dp_attention.py
@@ -143,7 +143,7 @@ def memcpy_triton_kernel(
    src_ptr,
    offset_ptr,
    sz_ptr,
-    offset_src,
+    offset_src: tl.constexpr,
    chunk_size,  # multiplied for offset and sz
    BLOCK_SIZE: tl.constexpr,
 ):