From fe712aa3df4876aa11ea923ed20ed1388f962e45 Mon Sep 17 00:00:00 2001 From: Xuchun Shang Date: Wed, 24 Dec 2025 10:31:21 +0800 Subject: [PATCH] [bug fix] fix hicache jit kernel (#15177) Signed-off-by: Xuchun Shang --- python/sglang/jit_kernel/hicache.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/jit_kernel/hicache.py b/python/sglang/jit_kernel/hicache.py index 1d015fe00..547087340 100644 --- 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"), ], )