github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
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