ColossalAI/tests/test_utils
ver217 26b7aac0be
[zero] reorganize zero/gemini folder structure (#3424)
* [zero] refactor low-level zero folder structure

* [zero] fix legacy zero import path

* [zero] fix legacy zero import path

* [zero] remove useless import

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor legacy zero import path

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor legacy zero import path

* [zero] fix test import path

* [zero] fix test

* [zero] fix circular import

* [zero] update import
2023-04-04 13:48:16 +08:00
..
test_checkpoint [hotfix] skipped unsafe test cases (#1282) 2022-07-13 00:08:59 +08:00
test_checkpoint_io [CheckpointIO] a uniform checkpoint I/O module (#1689) 2022-11-08 15:15:13 +08:00
test_lazy_init [hotfix] skip torchaudio tracing test (#3211) 2023-03-24 12:15:33 +08:00
test_activation_checkpointing.py [utils] Add use_reetrant=False in utils.activation_checkpoint (#1460) 2022-08-16 15:39:20 +08:00
test_colo_checkpoint.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
test_commons.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
test_flash_attention.py updated flash attention usage 2023-03-20 17:57:04 +08:00
test_lazy_init_ctx.py [utils] integrated colotensor with lazy init context (#1324) 2022-07-15 17:47:12 +08:00
test_memory.py [test] ignore 8 gpu test (#1080) 2022-06-08 23:14:18 +08:00
test_norm_gradient_clipping.py [Doc] add more doc for ColoTensor. (#1458) 2022-08-16 10:38:41 +08:00
test_zero_gradient_clippling.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00