ColossalAI/colossalai/booster/plugin
Baizhou Zhang 0bb0b481b4 [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
..
__init__.py
dp_plugin_base.py
gemini_plugin.py [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
low_level_zero_plugin.py
plugin_base.py
torch_ddp_plugin.py
torch_fsdp_plugin.py