ColossalAI/colossalai/fx/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
..
__init__.py
activation_checkpoint_codegen.py [fx] Add nested checkpoint in activation checkpoint codegen (#1585) 2022-09-12 20:00:48 +08:00