Boyuan Yao 1f2e547f7a [fx] Fix ckpt functions' definitions in forward (#1476) ...

* [fx] fix defining ckpt functions inside forward

* [fx] Modify activation checkpoint codegen and add ColoGraphModule

* [fx] some modification

* some modifications

* some modifications

* some modifications

* some modifications

* some code modifications

2022-08-22 16:59:54 +08:00

..

__init__.py

[fx] added activation checkpoint codegen support for torch < 1.12 (#1359)

2022-07-25 23:35:31 +08:00

activation_checkpoint_codegen.py

[fx] Fix ckpt functions' definitions in forward (#1476)

2022-08-22 16:59:54 +08:00