[bug fix] fix hicache jit kernel (#15177)

Signed-off-by: Xuchun Shang <xuchun.shang@gmail.com>
2025-12-24 10:31:21 +08:00
parent 846953d9f1
commit fe712aa3df
1 changed files with 2 additions and 2 deletions
--- a/python/sglang/jit_kernel/hicache.py
+++ b/python/sglang/jit_kernel/hicache.py
@@ -28,8 +28,8 @@ def _jit_hicache_module(*, element_size: int, unroll: int, block_quota: int) ->
        *args,
        cuda_files=["hicache.cuh"],
        cuda_wrappers=[
-            ("launch_one", f"HiCacheKernel<{args}>::run_one"),
-            ("launch_all", f"HiCacheKernel<{args}>::run_all"),
+            ("launch_one", f"&HiCacheKernel<{args}>::run_one"),
+            ("launch_all", f"&HiCacheKernel<{args}>::run_all"),
        ],
    )