[Feat] upgrade pytorch2.6 (#5417)

2025-04-20 16:06:34 -07:00
parent 475e2e378a
commit 417b44eba8
7 changed files with 8 additions and 8 deletions
@@ -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,
 ):