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