Hongxin Liu ffffc32dc7 [checkpointio] fix gemini and hybrid parallel optim checkpoint (#5347) ...

* [checkpointio] fix hybrid parallel optim checkpoint

* [extension] fix cuda extension

* [checkpointio] fix gemini optimizer checkpoint

* polish code

2024-02-01 16:13:06 +08:00

..

test_gemini_checkpoint_io.py

[checkpointio] fix gemini and hybrid parallel optim checkpoint (#5347)

2024-02-01 16:13:06 +08:00

test_gemini_torch_compability.py

[workflow] fixed build CI (#5240)

2024-01-10 22:34:16 +08:00

test_general_checkpoint_io.py

[hotfix] fix lr scheduler bug in torch 2.0 (#4864)

2023-10-12 14:04:24 +08:00

test_hybrid_parallel_plugin_checkpoint_io.py

[checkpointio] fix gemini and hybrid parallel optim checkpoint (#5347)

2024-02-01 16:13:06 +08:00

test_low_level_zero_checkpoint_io.py

[Pipeline Inference] Sync pipeline inference branch to main (#4820)

2023-10-11 11:40:06 +08:00

test_plugins_huggingface_compatibility.py

[workflow] fixed build CI (#5240)

2024-01-10 22:34:16 +08:00

test_torch_ddp_checkpoint_io.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

test_torch_fsdp_checkpoint_io.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

utils.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00