Boyuan Yao 092b9c8f49 [fx] Add use_reentrant=False to checkpoint in codegen (#1463) ...

* [utils] Add use_reetrant=False into colossalai checkpoint

* [utils] add some annotation in utils.activaion_checkpoint

* [test] add reset_seed at the beginning of tests in test_actiavion_checkpointing.py

* [test] modify test_activation_checkpoint.py

* [test] modify test for reentrant=False

* [fx] Add use_reentrant=False of checkpoint into codegen

2022-08-17 10:34:50 +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 use_reentrant=False to checkpoint in codegen (#1463)

2022-08-17 10:34:50 +08:00