You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_utils
Hongxin Liu 554aa9592e
[legacy] move communication and nn to legacy and refactor logger (#4671)
1 year ago
..
test_checkpoint [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
test_activation_checkpointing.py [devops] add large-scale distributed test marker (#4452) 1 year ago
test_commons.py [test] refactor tests with spawn (#3452) 2 years ago
test_flash_attention.py [shardformer] update shardformer to use flash attention 2 (#4392) 1 year ago
test_memory.py [test] refactor tests with spawn (#3452) 2 years ago
test_norm_gradient_clipping.py [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_zero_gradient_clippling.py [test] refactor tests with spawn (#3452) 2 years ago