54 Commits

Author	SHA1	Message	Date
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
Frank Lee	a9b8402d93	[booster] added the accelerator implementation (#3159)	2023-03-20 13:59:24 +08:00
Frank Lee	1ad3a636b1	[test] fixed torchrec model test (#3167) ... * [test] fixed torchrec model test * polish code * polish code * polish code * polish code * polish code * polish code	2023-03-20 11:40:25 +08:00
Frank Lee	ed19290560	[booster] implemented mixed precision class (#3151) ... * [booster] implemented mixed precision class * polish code	2023-03-17 11:00:15 +08:00