ColossalAI/tests/test_zero/test_legacy
Frank Lee 80eba05b0a
[test] refactor tests with spawn (#3452)
* [test] added spawn decorator

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code
2023-04-06 14:51:35 +08:00
..
common.py [test] reorganize zero/gemini tests (#3445) 2023-04-06 09:38:25 +08:00
test_found_inf.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_gemini_manager.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_init_context.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_param_op.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_shard_model_v2.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_shard_param.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_sharded_optim_state_dict.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_sharded_optim_v2.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_sharded_optim_with_sync_bn.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_state_dict.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_tensor_utils.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_zero_engine.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00