[bugfix]: fix modular ops activation checkpoint recomputation - #1716

Open

shaoxiongduan wants to merge 3 commits into

hao-ai-lab:mainfrom

shaoxiongduan:shao/fix-selective-activation-checkpointing

Commits on Aug 26, 2026

[bugfix]: retain fused attention in modular ops checkpointing

shaoxiongduan
authored and
SolitaryThinker
committed
40c1ca8
[bugfix]: close selective checkpoint policy gaps
SolitaryThinker
committed
20e487e
[bugfix]: finish modular checkpoint integration
SolitaryThinker
committed
cce7fdf