Frank Lee 7cfed5f076 [feat] refactored extension module (#5298) ...

* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

2024-01-25 17:01:48 +08:00

..

test_gemini_checkpoint_io.py

[feat] refactored extension module (#5298)

2024-01-25 17:01:48 +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

[ci] fix test_hybrid_parallel_plugin_checkpoint_io.py (#5276)

2024-01-17 13:38:55 +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