[Refactor] Switch gradient checkpointing to the non-reentrant implementation (1/3) - #1979

Open

HAOCHENYE wants to merge 7 commits into

mainfrom

feat/sac-nonlegacy-base

Commits on Jul 29, 2026

[Refactor] Switch gradient checkpointing to the non-reentrant implementation

HAOCHENYE
authored and
jayhenry
committed
[Feature] Add the shared contract for region-level selective checkpointing

HAOCHENYE
authored and
jayhenry
committed
[Fix] Make checkpoint_record safe inside fullgraph-compiled forwards

HAOCHENYE
authored and
jayhenry
committed
[Fix] Keep MTP gradients and the DSA top-k lifecycle correct under the new checkpointing

HAOCHENYE
authored and
jayhenry
committed
[Fix] Correct the reason `context_fn` is omitted when unset

HAOCHENYE
authored and
jayhenry
committed
[Refactor] Drop the legacy reentrant checkpoint path

HAOCHENYE
authored and
jayhenry
committed
[Refactor] Remove the now-unused DSA lifecycle predicate

HAOCHENYE
authored and
jayhenry
committed