ColossalAI/tests/test_utils
HELSON e5d615aeee
[hotfix] fix bugs in testing (#659)
* remove hybrid adam in test_moe_zero_optim

* fix activation checkpointing and its unitest
2022-04-02 21:58:47 +08:00
..
test_checkpoint
test_activation_checkpointing.py [hotfix] fix bugs in testing (#659) 2022-04-02 21:58:47 +08:00
test_bucket_tensor_copy.py
test_commons.py
test_gradient_accumluation.py
test_tensor_move.py
test_zero_gradient_clippling.py