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

6.7 KiB

Raw Blame History

View Raw