ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-25 05:22:43 +00:00

Files

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

2023-04-06 09:38:25 +08:00

test_found_inf.py

2023-04-06 14:51:35 +08:00

test_gemini_manager.py

2023-04-06 14:51:35 +08:00

test_init_context.py

2023-04-06 14:51:35 +08:00

test_param_op.py

2023-04-06 14:51:35 +08:00

test_shard_model_v2.py

2023-04-06 14:51:35 +08:00

test_shard_param.py

2023-04-06 14:51:35 +08:00

test_sharded_optim_state_dict.py

2023-04-06 14:51:35 +08:00

test_sharded_optim_v2.py

2023-04-06 14:51:35 +08:00

test_sharded_optim_with_sync_bn.py

2023-04-06 14:51:35 +08:00

test_state_dict.py

2023-04-06 14:51:35 +08:00

test_tensor_utils.py

2023-04-06 14:51:35 +08:00

test_zero_engine.py

2023-04-06 14:51:35 +08:00