Boyuan Yao f3687e4ee2 [fx] Add nested checkpoint in activation checkpoint codegen (#1585) ...

* [fx] add nested activation_checkpoint codegen

* undo algorithms commits

* solver

* undo some commits

* [fx] torch11 add nested activation checkpoint codegen

* remove some imports

* [fx] add some comments in activation codegen

* [fx] codegen instance error fix

2022-09-12 20:00:48 +08:00

..

test_activation_checkpoint_codegen.py

[fx] Fix activation codegen dealing with checkpointing first op (#1510)

2022-08-27 19:39:21 +08:00

test_nested_activation_checkpoint_codegen.py

[fx] Add nested checkpoint in activation checkpoint codegen (#1585)

2022-09-12 20:00:48 +08:00