ColossalAI/tests/test_utils
Hongxin Liu dbb32692d2
[lazy] refactor lazy init (#3891)
* [lazy] remove old lazy init

* [lazy] refactor lazy init folder structure

* [lazy] fix lazy tensor deepcopy

* [test] update lazy init test
2023-06-05 14:20:47 +08:00
..
test_checkpoint
test_checkpoint_io
test_activation_checkpointing.py
test_colo_checkpoint.py
test_commons.py
test_flash_attention.py
test_memory.py
test_norm_gradient_clipping.py
test_zero_gradient_clippling.py