Repository navigation

perf(cake_kda): faster context-parallel host path, wave-fill plan rule and capped graph replay, with grad- #7089 - #7097

Merged

yyihuang merged 15 commits into

flashinfer-ai:mainfrom

yyihuang:cake-kda-train-kcp-pr3

Oct 23, 2026

Merged

perf(cake_kda): faster context-parallel host path, wave-fill plan rule and capped graph replay, with grad- #7089#7097
yyihuang merged 15 commits into
flashinfer-ai:mainfrom
yyihuang:cake-kda-train-kcp-pr3

Commits on Oct 23, 2026
- cake_kda_train: trimmed host path, P = 1 tie plan rule and kernel-node replay cap for the context-parallel host
  
  averyhNV
  and
  claude
  committed
  ea629d2
- cake_causal_conv1d_train: trimmed host path of the context-parallel halo calls
  
  averyhNV
  and
  claude
  committed
  e2bae53
- test: cover the context-parallel KDA host's trimmed path, tie plan rule, fused carry merges and kernel-node replay cap
  
  averyhNV
  and
  claude
  committed
  19be574
- cake_kda_train: document the trimmed host path, the P = 1 tie rule and the capped graph replay
  
  averyhNV
  and
  claude
  committed
  5e9773c
- test: compare the KCP manifests' generator records by form
  
  averyhNV
  and
  claude
  committed
  7b885b3
- test: name the base plan rule explicitly in the P = 1 tie row enumeration
  
  averyhNV
  and
  claude
  committed
  73d8eb0
- cake_kda_train: keep the KCP segment table its own object
  
  averyhNV
  and
  claude
  committed
  5918500
- cake_kda_train: run the KCP local stages on the current program set
  
  averyhNV
  and
  claude
  committed
  659ac18
- cake_kda_train: stop saving the token-row Aqk / Akk under KCP
  
  averyhNV
  and
  claude
  committed
  e54334b
- cake_kda_train: fix stale plan-rule wording in the context-parallel docs
  
  averyhNV
  and
  claude
  committed
  10be4ac
- cake_kda_train: ship the context-parallel output and adjoint programs
  
  averyhNV
  and
  claude
  committed
  fb85490
- test: cover the context-parallel output and adjoint records in the build test
  
  averyhNV
  and
  claude
  committed
  60b4362
- cake_kda_train: document the two gate-row forms the KCP walks read
  
  averyhNV
  and
  claude
  committed
  7e941c4
- cake_kda_train: name the knobs instead of their internal labels in the context-parallel comments
  
  averyhNV
  and
  claude
  committed
  082781b
- cake_kda_train: say what the two unsupported context-parallel options are
  
  averyhNV
  and
  claude
  committed
  bd9b839

Commits

Commits on Oct 23, 2026