ColossalAI/colossalai/booster/plugin
Hongxin Liu 6552cbf8e1
[booster] fix no_sync method (#3709)
* [booster] fix no_sync method

* [booster] add test for ddp no_sync

* [booster] fix merge

* [booster] update unit test

* [booster] update unit test

* [booster] update unit test
2023-05-09 11:10:02 +08:00
..
__init__.py [booster] add low level zero plugin (#3594) 2023-04-26 14:37:25 +08:00
dp_plugin_base.py [booster] update prepare dataloader method for plugin (#3706) 2023-05-08 15:44:03 +08:00
gemini_plugin.py [booster] fix no_sync method (#3709) 2023-05-09 11:10:02 +08:00
low_level_zero_plugin.py [booster] fix no_sync method (#3709) 2023-05-09 11:10:02 +08:00
plugin_base.py [booster] fix no_sync method (#3709) 2023-05-09 11:10:02 +08:00
torch_ddp_plugin.py [booster] fix no_sync method (#3709) 2023-05-09 11:10:02 +08:00