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