Xuchun Shang 3d68bd9d9b add hicache jit test (#17847) ...

Signed-off-by: Xuchun Shang <xuchun.shang@linux.alibaba.com>

2026-02-06 16:54:33 +08:00

..

benchmark

add hicache jit test (#17847)

2026-02-06 16:54:33 +08:00

csrc

[Kernel] Migrate GPTQ-Marlin GEMM kernel to JIT (#18067)

2026-02-06 08:31:42 +08:00

diffusion/cutedsl

[diffusion] fix: fix torch.compile graph break caused by torch._dynamo.disable (#18336)

2026-02-06 14:48:09 +08:00

flash_attention/cute

[Diffusion] Fix Ring Parallel bug with FA4 (#18062)

2026-02-02 17:06:51 +08:00

include/sgl_kernel

[Kernel] Migrate GPTQ-Marlin GEMM kernel to JIT (#18067)

2026-02-06 08:31:42 +08:00

tests

[diffusion] fix: fix torch.compile graph break caused by torch._dynamo.disable (#18336)

2026-02-06 14:48:09 +08:00

__init__.py

Move fa4 from sgl-kernel to jit kernel (#17353)

2026-01-24 15:25:03 +08:00

__main__.py

[Minor] Enhance JIT kernel and add dev docs (#14570)

2025-12-23 22:34:59 +08:00

.clang-format

[Refactor] Clean up JIT kernel utilites (#16884)

2026-01-13 17:54:16 +08:00

add_constant.py

[JIT kernel] Update jit_kernel cache and develop doc (#17842)

2026-01-28 15:09:47 +08:00

concat_mla.py

[Move sgl-kernel Kernel to JIT] Add JIT concat MLA kernels (#17889)

2026-02-03 10:49:17 +08:00

cuda_wait_value.py

[JIT kernel] Update jit_kernel cache and develop doc (#17842)

2026-01-28 15:09:47 +08:00

cutedsl_gdn.py

[jit-kernel] Add CuTe DSL GDN Decode Kernel (#15631)

2026-01-18 12:54:36 -08:00

flash_attention_v4.py

Move fa4 from sgl-kernel to jit kernel (#17353)

2026-01-24 15:25:03 +08:00

gptq_marlin.py

[Kernel] Migrate GPTQ-Marlin GEMM kernel to JIT (#18067)

2026-02-06 08:31:42 +08:00

hicache.py

[JIT kernel] Update jit_kernel cache and develop doc (#17842)

2026-01-28 15:09:47 +08:00

kvcache.py

[Feature] Support JIT set kv cache (#16273)

2026-01-10 17:34:09 -08:00

norm.py

[Diffsuion & JIT_kernel] QKNorm cross heads kernel (#18073)

2026-02-03 10:03:17 +08:00

per_tensor_quant_fp8.py

[Refactor] Clean up JIT kernel utilites (#16884)

2026-01-13 17:54:16 +08:00

rope.py

[Kernel] Add JIT apply_rope_with_cos_sin_cache_inplace (#18155)

2026-02-05 21:49:37 +08:00

timestep_embedding.py

[JIT kernel] Update jit_kernel cache and develop doc (#17842)

2026-01-28 15:09:47 +08:00

utils.py

[JIT kernel] Update jit_kernel cache and develop doc (#17842)

2026-01-28 15:09:47 +08:00