ColossalAI/tests/test_utils
Hongxin Liu 26e29d58f0
[devops] add large-scale distributed test marker (#4452)
* [test] remove cpu marker

* [test] remove gpu marker

* [test] update pytest markers

* [ci] update unit test ci
2023-08-16 18:56:52 +08:00
..
test_checkpoint [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_activation_checkpointing.py [devops] add large-scale distributed test marker (#4452) 2023-08-16 18:56:52 +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 [shardformer] update shardformer to use flash attention 2 (#4392) 2023-08-15 23:25:14 +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