mirror of https://github.com/hpcaitech/ColossalAI
26b7aac0be
* [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 |
||
---|---|---|
.. | ||
test_checkpoint | ||
test_checkpoint_io | ||
test_lazy_init | ||
test_activation_checkpointing.py | ||
test_colo_checkpoint.py | ||
test_commons.py | ||
test_flash_attention.py | ||
test_lazy_init_ctx.py | ||
test_memory.py | ||
test_norm_gradient_clipping.py | ||
test_zero_gradient_clippling.py |