github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-01 20:05:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
4fb4a22a72
ColossalAI/tests/test_booster/test_plugin
History
flybird11111 2a2ec49aa7
[plugin]fix 3d checkpoint load when booster boost without optimizer. (#5135) 
* fix 3d checkpoint load when booster boost without optimizer

fix 3d checkpoint load when booster boost without optimizer

* test ci

* revert ci

* fix

fix
2023-11-30 18:37:47 +08:00
..
test_3d_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_dp_plugin_base.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_gemini_plugin.py [plugin]fix 3d checkpoint load when booster boost without optimizer. (#5135) 2023-11-30 18:37:47 +08:00
test_low_level_zero_plugin.py [npu] add npu support for gemini and zero (#5067) 2023-11-20 16:12:41 +08:00
test_torch_ddp_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_torch_fsdp_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00