Extend CheckpointFunction to track all tensor input/output - #1148

Open

000Justin000 wants to merge 3 commits into

facebookresearch:mainfrom

000Justin000:checkpointing

Commits on Nov 7, 2023

Extend CheckpointFunction to track all tensor input/output
Junteng Jia
committed
a93b025
Extend CheckpointFunction to track all tensor input/output (add comments)
Junteng Jia
committed
55738b5
Extend CheckpointFunction to track all tensor input/output (add comments)
Junteng Jia
committed
8710f03