mirror of https://github.com/hpcaitech/ColossalAI
092b9c8f49
* [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 |
||
|---|---|---|
| .. | ||
| codegen | ||
| passes | ||
| tracer | ||
| __init__.py | ||
| proxy.py | ||