Yuan Luo 55444ed667 [EP] Add cuda kernel for moe_ep_pre_reorder (#6699) ...

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>

2025-06-01 20:49:01 -07:00

..

__init__.py

[EP] Add cuda kernel for moe_ep_pre_reorder (#6699)

2025-06-01 20:49:01 -07:00

allreduce.py

sgl-kernel transfer custom allreduce from trt kernel to vllm kernel (#5079)

2025-04-05 14:23:20 -07:00

attention.py

Add Cutlass MLA attention backend (#5390)

2025-04-27 20:58:53 -07:00

elementwise.py

[Feat] Enable PDL automatically on Hopper architecture (#5981)

2025-06-01 12:30:17 -07:00

flash_attn.py

Revert "fix some typos" (#6244)

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

gemm.py

[1/2] Support Qserve (#6457)

2025-05-21 19:48:59 -07:00

grammar.py

fix sgl-kernel unit tests (#5666)

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

moe.py

[EP] Add cuda kernel for moe_ep_pre_reorder (#6699)

2025-06-01 20:49:01 -07:00

sampling.py

Fix sampler nan check when calling top_k_top_p_sampling_from_probs (#5546)

2025-04-19 21:47:23 -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

speculative.py

use default for torch.ops (#4835)

2025-03-27 19:09:58 -07:00

utils.py

misc: cache is_hopper_arch (#6799)

2025-06-01 15:28:31 -07:00

version.py

chore: bump sgl-kernel v0.1.5 (#6794)

2025-05-31 14:54:00 -07:00