github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 14:43:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1653063fce7062dfd652e73691fac44a24698cd5
ColossalAI/colossalai/booster/plugin
History
Frank Lee e7f3bed2d3 [booster] added the plugin base and torch ddp plugin (#3180) 
* [booster] added the plugin base and torch ddp plugin

* polish code

* polish code

* polish code
2023-03-21 17:39:30 +08:00
..
__init__.py
[booster] added the plugin base and torch ddp plugin (#3180)
2023-03-21 17:39:30 +08:00
plugin_base.py
[booster] added the plugin base and torch ddp plugin (#3180)
2023-03-21 17:39:30 +08:00
torch_ddp_plugin.py
[booster] added the plugin base and torch ddp plugin (#3180)
2023-03-21 17:39:30 +08:00