[Refactor] Deduplicate NSA utils.py into cp_utils.py for context parallel#22914

Merged

Fridge003 merged 7 commits intomainfrom

dedup-nsa-utils-cp-utils

Apr 20, 2026

Commits on Apr 15, 2026

[Refactor] Deduplicate NSA utils.py into cp_utils.py for context parallel

Fridge003
and
claude
committed

Commits on Apr 16, 2026

some fixes
Fridge003
committed

Commits on Apr 17, 2026

Fix prefix_len double-count in _get_topk_ragged_with_cp

Fridge003
and
claude
committed

Commits on Apr 18, 2026

Keep extend-only kv_len_prev/next so NSA CP handles multi-batch extend
Fridge003
committed
Gate extend-only kv_len_prev/next on NSA; keep prefix-baked path for non-NSA CP
Fridge003
committed

Commits on Apr 20, 2026