ColossalAI/colossalai/booster/plugin
Hongxin Liu fae6c92ead
Merge branch 'main' into feature/shardformer
2023-09-05 21:54:08 +08:00
..
__init__.py [plugin] add 3d parallel plugin (#4295) 2023-08-15 23:25:14 +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 Merge branch 'main' into feature/shardformer 2023-09-04 23:43:13 +08:00
hybrid_parallel_plugin.py [shardformer] Add overlap optional for HybridParallelPlugin (#4615) 2023-09-05 11:52:23 +08:00
low_level_zero_plugin.py [zero] hotfix master param sync (#4618) 2023-09-05 15:04:02 +08:00
plugin_base.py [zero]support no_sync method for zero1 plugin (#4138) 2023-07-31 22:13:29 +08:00
pp_plugin_base.py [plugin] add 3d parallel plugin (#4295) 2023-08-15 23:25:14 +08:00
torch_ddp_plugin.py [zero]support no_sync method for zero1 plugin (#4138) 2023-07-31 22:13:29 +08:00
torch_fsdp_plugin.py [zero]support no_sync method for zero1 plugin (#4138) 2023-07-31 22:13:29 +08:00