github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_utils
History
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] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_checkpoint_io [CI] fix some spelling errors (#3707) 2023-05-10 17:12:03 +08:00
test_activation_checkpointing.py [CI] fix some spelling errors (#3707) 2023-05-10 17:12:03 +08:00
test_colo_checkpoint.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_commons.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_flash_attention.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_memory.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_norm_gradient_clipping.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_zero_gradient_clippling.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00