[test] refactor tests with spawn (#3452)

* [test] added spawn decorator * polish code * polish code * polish code * polish code * polish code * polish code

[test] refactor tests with spawn (#3452)
* [test] added spawn decorator * polish code * polish code * polish code * polish code * polish code * polish code
80eba05b · Frank Lee · GitHub · 62f4e2eb · 80eba05b · 80eba05b
Unverified Commit 80eba05b authored Apr 06, 2023 by Frank Lee Committed by GitHub Apr 06, 2023
20 changed files
--- a/tests/test_zero/test_gemini/test_runtime_mem_tracer.py
+++ b/tests/test_zero/test_gemini/test_runtime_mem_tracer.py
--- a/tests/test_zero/test_gemini/test_search.py
+++ b/tests/test_zero/test_gemini/test_search.py
--- a/tests/test_zero/test_gemini/test_zeroddp_state_dict.py
+++ b/tests/test_zero/test_gemini/test_zeroddp_state_dict.py
--- a/tests/test_zero/test_gemini/test_zerooptim_state_dict.py
+++ b/tests/test_zero/test_gemini/test_zerooptim_state_dict.py
--- a/tests/test_zero/test_legacy/test_found_inf.py
+++ b/tests/test_zero/test_legacy/test_found_inf.py
--- a/tests/test_zero/test_legacy/test_gemini_manager.py
+++ b/tests/test_zero/test_legacy/test_gemini_manager.py
--- a/tests/test_zero/test_legacy/test_init_context.py
+++ b/tests/test_zero/test_legacy/test_init_context.py
--- a/tests/test_zero/test_legacy/test_param_op.py
+++ b/tests/test_zero/test_legacy/test_param_op.py
--- a/tests/test_zero/test_legacy/test_shard_model_v2.py
+++ b/tests/test_zero/test_legacy/test_shard_model_v2.py
--- a/tests/test_zero/test_legacy/test_shard_param.py
+++ b/tests/test_zero/test_legacy/test_shard_param.py
--- a/tests/test_zero/test_legacy/test_sharded_optim_state_dict.py
+++ b/tests/test_zero/test_legacy/test_sharded_optim_state_dict.py
--- a/tests/test_zero/test_legacy/test_sharded_optim_v2.py
+++ b/tests/test_zero/test_legacy/test_sharded_optim_v2.py
--- a/tests/test_zero/test_legacy/test_sharded_optim_with_sync_bn.py
+++ b/tests/test_zero/test_legacy/test_sharded_optim_with_sync_bn.py
--- a/tests/test_zero/test_legacy/test_state_dict.py
+++ b/tests/test_zero/test_legacy/test_state_dict.py
--- a/tests/test_zero/test_legacy/test_tensor_utils.py
+++ b/tests/test_zero/test_legacy/test_tensor_utils.py
--- a/tests/test_zero/test_legacy/test_zero_engine.py
+++ b/tests/test_zero/test_legacy/test_zero_engine.py
--- a/tests/test_zero/test_low_level/test_grad_acc.py
+++ b/tests/test_zero/test_low_level/test_grad_acc.py
--- a/tests/test_zero/test_low_level/test_zero1_2.py
+++ b/tests/test_zero/test_low_level/test_zero1_2.py
--- a/tests/test_zero/test_low_level/test_zero_init.py
+++ b/tests/test_zero/test_low_level/test_zero_init.py
--- a/tests/test_zero/test_low_level/test_zero_tp.py
+++ b/tests/test_zero/test_low_level/test_zero_tp.py