Hongxin Liu 3c07a2846e [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) ... * [test] refactor torch ddp checkpoint test * [plugin] update low level zero optim checkpoint * [plugin] update gemini optim checkpoint		2023-05-19 19:42:31 +08:00
..
components_to_test	[CI] fix typo with tests/ etc. (#3727)	2023-05-11 16:30:58 +08:00
kit	[booster] add low level zero plugin (#3594)	2023-04-26 14:37:25 +08:00
test_amp	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_analyzer	[devops] update torch version of CI (#3725)	2023-05-15 17:20:56 +08:00
test_auto_parallel	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_autochunk	[devops] update torch version of CI (#3725)	2023-05-15 17:20:56 +08:00
test_booster	[plugin] torch ddp plugin supports sharded model checkpoint (#3775)	2023-05-18 20:05:59 +08:00
test_checkpoint_io	[plugin] a workaround for zero plugins' optimizer checkpoint (#3780)	2023-05-19 19:42:31 +08:00
test_cluster	[devops] update torch version of CI (#3725)	2023-05-15 17:20:56 +08:00
test_comm	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_config	[pipeline] refactor the pipeline module (#1087)	2022-06-10 11:27:38 +08:00
test_context	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_data	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_data_pipeline_tensor_parallel	[CI] fix typo with tests/ etc. (#3727)	2023-05-11 16:30:58 +08:00
test_ddp	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_device	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_engine	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_fx	[devops] update torch version of CI (#3725)	2023-05-15 17:20:56 +08:00
test_layers	[CI] fix typo with tests/ etc. (#3727)	2023-05-11 16:30:58 +08:00
test_moe	[CI] fix typo with tests/ etc. (#3727)	2023-05-11 16:30:58 +08:00
test_ops	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_optimizer	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_pipeline	[test] refactor tests with spawn (#3452)	2023-04-06 14:51:35 +08:00
test_tensor	[CI] fix typo with tests/ etc. (#3727)	2023-05-11 16:30:58 +08:00
test_trainer	[CI] fix typo with tests/ etc. (#3727)	2023-05-11 16:30:58 +08:00
test_utils	[devops] update torch version of CI (#3725)	2023-05-15 17:20:56 +08:00
test_zero	[CI] fix typo with tests/ etc. (#3727)	2023-05-11 16:30:58 +08:00
__init__.py	[zero] Update sharded model v2 using sharded param v2 (#323)	2022-03-11 15:50:28 +08:00