Repository navigation

[Feat] kimi-k3.1: DSPARK and DFLASH under decode context parallelism - #48739

Open

kpham-sgl wants to merge 31 commits into

sgl-project:mainfrom

kpham-sgl:kpham/k31-dcp-spec-verify

Open

[Feat] kimi-k3.1: DSPARK and DFLASH under decode context parallelism#48739
kpham-sgl wants to merge 31 commits into
sgl-project:mainfrom
kpham-sgl:kpham/k31-dcp-spec-verify

Commits on Nov 5, 2026
- [Feat] kimi-k3.1: Raw port of the DSA DCP decode runtime from #48414
  
  committed
  340e28e
- [Feat] kimi-k3.1: Adapt the DCP runtime to main's DSA kernel APIs
  kpham-sgl
  committed
  a30510d
- [Feat] kimi-k3.1: Limit the DCP port to plain decode on flashmla_kv
  kpham-sgl
  committed
  88f591e
- [Chore] dsa: Import FlashMLASchedMeta from flash_mla
  kpham-sgl
  committed
  5649848
- [Feat] kimi-k3.1: Raw port of DSPARK and DFLASH under DCP from #48414
  
  yhyang201
  authored and
  kpham-sgl
  committed
  517af66
- [Feat] dflash: Detect MLA drafts with main's parser; admit only cutedsl_mla
  kpham-sgl
  committed
  307a7b0
- [Fix] dflash: Raw port of the DCP draft KV budget from #48414
  
  yhyang201
  authored and
  kpham-sgl
  committed
  0f6d2b4
- [Fix] dflash: Charge the DCP draft KV by the draft's loc_space_scale
  kpham-sgl
  committed
  72f5388
- [Chore] kimi-k3.1: Drop verify-only DCP leftovers and trim comments
  kpham-sgl
  committed
  76ccc48
- [Chore] kimi-k3.1: Remove restating comments from the DCP port
  kpham-sgl
  committed
  652bf10
- [Chore] kimi-k3.1: Resolve the DCP config inline, as Kimi-K3 does
  kpham-sgl
  committed
  1699d9a
- [Chore] kimi-k3.1: Key the sharded index-K flag on is_kimi_k31
  kpham-sgl
  committed
  458e5da
- [Chore] kimi-k3.1: Explain the sharded index-K decision in one place
  kpham-sgl
  committed
  26ce335
- [Chore] kimi-k3.1: Drop the no-op save_kv_cache argument
  kpham-sgl
  committed
  4818f4c
- [Refactor] kimi-k3.1: Move the DCP prefill KV gather out of the DSA backend
  kpham-sgl
  committed
  2d3e598
- Revert "[Refactor] kimi-k3.1: Move the DCP prefill KV gather out of the DSA backend"
  kpham-sgl
  committed
  4358e9f
- [Chore] kimi-k3.1: Let FlashMLA schedule DCP decode itself
  kpham-sgl
  committed
  29117fd
Commits on Nov 6, 2026

Navigation Menu

Repository navigation

[Feat] kimi-k3.1: DSPARK and DFLASH under decode context parallelism - #48739

[Feat] kimi-k3.1: DSPARK and DFLASH under decode context parallelism#48739
kpham-sgl wants to merge 31 commits into
sgl-project:mainfrom
kpham-sgl:kpham/k31-dcp-spec-verify

Commits on Nov 5, 2026

[Feat] kimi-k3.1: Raw port of the DSA DCP decode runtime from #48414

[Feat] kimi-k3.1: Adapt the DCP runtime to main's DSA kernel APIs

[Feat] kimi-k3.1: Limit the DCP port to plain decode on flashmla_kv

[Chore] dsa: Import FlashMLASchedMeta from flash_mla

[Feat] kimi-k3.1: Raw port of DSPARK and DFLASH under DCP from #48414

[Feat] dflash: Detect MLA drafts with main's parser; admit only cutedsl_mla

[Fix] dflash: Raw port of the DCP draft KV budget from #48414

[Fix] dflash: Charge the DCP draft KV by the draft's loc_space_scale

[Chore] kimi-k3.1: Drop verify-only DCP leftovers and trim comments

[Chore] kimi-k3.1: Remove restating comments from the DCP port

[Chore] kimi-k3.1: Resolve the DCP config inline, as Kimi-K3 does

[Chore] kimi-k3.1: Key the sharded index-K flag on is_kimi_k31

[Chore] kimi-k3.1: Explain the sharded index-K decision in one place

[Chore] kimi-k3.1: Drop the no-op save_kv_cache argument

[Refactor] kimi-k3.1: Move the DCP prefill KV gather out of the DSA backend

Revert "[Refactor] kimi-k3.1: Move the DCP prefill KV gather out of the DSA backend"

[Chore] kimi-k3.1: Let FlashMLA schedule DCP decode itself

Commits on Nov 6, 2026

[Chore] dsa: TODO to shard index-K under DCP by default

[Chore] kimi-k3.1: Inline the FlashMLA head padding

[Chore] kimi-k3.1: Check the DCP top-k requirements at config time

[Chore] dsa: One DCP width on the DSA backend

[Chore] dsa: Rename dcp_all_visible to dcp_select_all_slots

Merge branch 'main' into kpham/k31-dcp-decode-runtime

Merge remote-tracking branch 'kpham/kpham/k31-dcp-decode-runtime' into kpham/k31-dcp-spec-verify

[Fix] kimi-k3.1: Restore target verify under DCP on the reshaped runtime

[Test] dsa: Give the backend stubs the DCP fields the port added

Merge remote-tracking branch 'kpham/kpham/k31-dcp-decode-runtime' into kpham/k31-dcp-spec-verify

[Test] dsa: Give every backend stub the DCP fields

Merge branch 'kpham/k31-dcp-decode-runtime' of https://github.com/kpham-sgl/sglang into kpham/k31-dcp-spec-verify

[Test] dsa: Rename the trtllm padding stub's DCP fields

Merge branch 'kpham/k31-dcp-decode-runtime' of https://github.com/kpham-sgl/sglang into kpham/k31-dcp-spec-verify