zyksir 8e3797be1c support 1 shot allreduce in 1-node and 2-node using mscclpp (#6277)

2025-06-04 22:11:24 -07:00

..

__init__.py

[1/2] Add Kernel support for Cutlass based Fused FP4 MoE (#6093)

2025-06-02 13:48:03 -07:00

allreduce.py

support 1 shot allreduce in 1-node and 2-node using mscclpp (#6277)

2025-06-04 22:11:24 -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] Add Kernel support for Cutlass based Fused FP4 MoE (#6093)

2025-06-02 13:48:03 -07:00

grammar.py

fix sgl-kernel unit tests (#5666)

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

moe.py

Set num_fused_shared_experts as num_shared_experts when shared_experts fusion is not disabled (#6736)

2025-06-04 15:53:22 -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