Boyuan Yao d6b01feb66 [fx] Modify offload codegen (#1618) ...

* [fx] modify offload codegen

* [fx] remove repeated hook definitions

* [fx] modify offload test

2022-09-23 11:04:52 +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] Modify offload codegen (#1618)

2022-09-23 11:04:52 +08:00