Boyuan Yao a7cda6f57d [fx] Add offload codegen (#1598) ... * [fx] add input activation offload to codegen * [fx] modify unit test * [fx] remove two skips in torch11 * [fx] use all_input_nodes instead of _input_nodes		2022-09-14 15:49:06 +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] Add offload codegen (#1598)	2022-09-14 15:49:06 +08:00