Zhengbo Wang fb61164f27 [Refactor] Use is_in_ci() utility in JIT kernel benchmarks (#17118)

2026-01-25 20:40:47 +08:00

..

benchmark

[Refactor] Use is_in_ci() utility in JIT kernel benchmarks (#17118)

2026-01-25 20:40:47 +08:00

csrc

[Diffusion] Add diffusion time embedding to jit kernel (#17658)

2026-01-24 14:27:08 +08:00

flash_attention/cute

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

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

include/sgl_kernel

[Diffusion] Add diffusion time embedding to jit kernel (#17658)

2026-01-24 14:27:08 +08:00

tests

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

2026-01-24 15:25:03 +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

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

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

cuda_wait_value.py

Add cuda event based on waiting value (#14214)

2025-12-01 18:51:44 +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

hicache.py

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

2025-12-24 10:31:21 +08:00

kvcache.py

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

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

norm.py

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

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

per_tensor_quant_fp8.py

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

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

timestep_embedding.py

[Diffusion] Add diffusion time embedding to jit kernel (#17658)

2026-01-24 14:27:08 +08:00

utils.py

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

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