DarkSharpness 86a0be65d8 [Feature] Support custom set kv buffer kernel (#8884)

2025-08-12 16:56:51 -07:00

..

testing

Fuse writing KV buffer into rope kernel (part 1: sgl-kernel) (#9077)

2025-08-12 01:46:40 -07:00

__init__.py

[Feature] Support custom set kv buffer kernel (#8884)

2025-08-12 16:56:51 -07:00

allreduce.py

[Feature] Integrate quick allreduce and select the best allreduce implementation (#6619)

2025-07-24 20:48:42 -07:00

attention.py

[fix] fix cutlass_mla_backend with cuda_graph and add sm_scale for sgl-kernel cutlass_mla (#7184)

2025-06-14 12:45:41 -07:00

cutlass_moe.py

[1/n]: add cutlass W4A8 moe kernel for hopper architecture (#7772)

2025-07-04 20:50:12 -07:00

elementwise.py

Fuse writing KV buffer into rope kernel (part 1: sgl-kernel) (#9077)

2025-08-12 01:46:40 -07:00

flash_attn.py

Revert "fix some typos" (#6244)

2025-05-12 12:53:26 -07:00

fused_moe.py

refactor: Move scalar_types.py to sgl-kernel to avoid circular import (#8720)

2025-08-07 19:22:16 -07:00

gemm.py

Add bf16 output option for dsv3_router_gemm kernel (#7999)

2025-07-20 09:49:37 +08:00

grammar.py

fix sgl-kernel unit tests (#5666)

2025-04-23 01:18:30 -07:00

kvcacheio.py

fix page first per layer pf2lf kernel (#8915)

2025-08-09 17:16:11 -07:00

marlin.py

[1/n] apply wna16marlin kernel in moe weight only quantization (#7683)

2025-07-01 23:21:25 -07:00

memory.py

[Feature] Support custom set kv buffer kernel (#8884)

2025-08-12 16:56:51 -07:00

moe.py

Revert "[1/2][resubmit] sgl-kernel: Fuse routed scaling factor into m… (#9035)

2025-08-10 17:34:54 -07:00

sampling.py

Fix a bug in BatchTokenIDOut & Misc style and dependency updates (#7457)

2025-06-23 06:20:39 -07:00

scalar_type.py

refactor: Move scalar_types.py to sgl-kernel to avoid circular import (#8720)

2025-08-07 19:22:16 -07:00

sparse_flash_attn.py

[Feat] QWen-1M context support[1/2]: Update block sparse attention backend utils kernel (#5847)

2025-04-28 11:03:17 -07:00

spatial.py

[Bug] fix green context's incompatibility with cuda < 12.4 (#8701)

2025-08-02 15:23:11 -07:00

speculative.py

Add treemask mode to build_eagle_tree & release sgl-kernel 0.2.3 (#7756)

2025-07-05 12:17:05 -07:00

top_k.py

Fix sampling for speculative decoding & simplify kernels (#7207)

2025-06-16 03:28:30 -07:00

utils.py

Runtime check CUDA driver version to avoid unresolved green context symbols (#9021)

2025-08-12 09:26:10 -07:00

version.py

chore: bump sgl-kernel v0.3.4 (#9103)

2025-08-12 01:48:47 -07:00