DarkSharpness e0b2d3eebe [Feature] Add a fast-topk to sgl-kernel for DeepSeek v3.2 (#11194) ...

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>

2025-10-05 10:19:03 -07:00

..

testing

Fix FusedSetKVBufferArg in RotaryEmbedding (#11003)

2025-09-28 11:17:27 -07:00

__init__.py

[Feature] Add a fast-topk to sgl-kernel for DeepSeek v3.2 (#11194)

2025-10-05 10:19:03 -07:00

_fa4_interface.py

support using fa4 on deepseek on blackwell (#9928)

2025-09-16 16:16:06 -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

[1/2] Speed up trtllm_mla attention backend (>10% e2e) (#10473)

2025-09-15 11:53:21 -07:00

flash_attn.py

[1/2] Support FA4 for MHA Prefill in sgl-kernel (#10940)

2025-09-28 19:59:14 -07:00

fused_moe.py

Minor style fixes for sgl-kernel (#9289)

2025-08-18 09:38:35 -07:00

gemm.py

Revert "[1/2] Optimizations and refactors about quant kernel (#9534)" (#10292)

2025-09-10 18:24:23 -07:00

grammar.py

fix sgl-kernel unit tests (#5666)

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

kvcacheio.py

Page first direct IO kernel (#10060)

2025-09-10 13:35:34 +08:00

mamba.py

Add mamba kernel (#10234)

2025-09-09 12:58:43 -07:00

marlin.py

[2/n]decouple quantization implementation from vLLM dependency (#8112)

2025-08-14 03:19:03 -07:00

memory.py

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

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

moe.py

[sgl-kernel] Support moe_sum_reduce cuda kernel (#10321)

2025-09-19 14:12:09 +08:00

sampling.py

Minor style fixes for sgl-kernel (#9289)

2025-08-18 09:38:35 -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

Optional extension for green context (#9231)

2025-08-15 21:33:52 +08:00

speculative.py

[Feature] Speculative decoding support lookahead (#9873)

2025-09-18 16:42:41 -07:00

top_k.py

[Feature] Add a fast-topk to sgl-kernel for DeepSeek v3.2 (#11194)

2025-10-05 10:19:03 -07:00

utils.py

[fix] fix enable_pdl for blackwell (#9011)

2025-08-19 01:16:08 +08:00

version.py

chore: bump sgl-kernel v0.3.14 (#11067)

2025-09-30 02:53:24 -07:00