support checkpoint engine[P2P] recovery in colocate mode - #2027

Open

PengchengShi00 wants to merge 13 commits into

InternLM:mainfrom

PengchengShi00:checkpoint-engine-recovery

Commits on Aug 21, 2026

Fix bug: weight update flow when skip_load_weights
PengchengShi00
committed
fbbf31f
rename update_weights to weight_update
PengchengShi00
committed
f0b05d3
WIP:storce checkpoint-engine recovery
PengchengShi00
committed
ab1207d
fix pending parameter transfer
PengchengShi00
committed
ea33801
Fix pre-commit formatting issues
PengchengShi00
committed
7e5a6dd
fix unit test of recovery
PengchengShi00
committed
bbd5e88
remove state check in update weight
PengchengShi00
committed
a20a0f9
Add max pending task limit for async rollout production
PengchengShi00
committed
c323120
simplify worker recovery state transitions
PengchengShi00
committed
3f73e10
revert: remove temporary debug changes
PengchengShi00
committed
812b4c9
fail fast when checkpoint-engine register fails
PengchengShi00
committed
d2a3124
resolve rebase conflict
PengchengShi00
committed
e26209f
add shutdown before train and fix unit test
PengchengShi00
committed
5b31599