ColossalAI/tests/test_fx/test_codegen
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