106 Commits (5e1c93d732a3ee77657fe68c031dacdeae4d7807)

Author	SHA1	Message	Date
ver217	5f2e34e6c9	[booster] implement Gemini plugin (#3352) ... * [booster] add gemini plugin * [booster] update docstr * [booster] gemini plugin add coloparam convertor * [booster] fix coloparam convertor * [booster] fix gemini plugin device * [booster] add gemini plugin test * [booster] gemini plugin ignore sync bn * [booster] skip some model * [booster] skip some model * [booster] modify test world size * [booster] modify test world size * [booster] skip test	2023-03-31 16:06:13 +08:00
Frank Lee	73d3e4d309	[booster] implemented the torch ddd + resnet example (#3232) ... * [booster] implemented the torch ddd + resnet example * polish code	2023-03-27 10:24:14 +08:00
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	ed19290560	[booster] implemented mixed precision class (#3151) ... * [booster] implemented mixed precision class * polish code	2023-03-17 11:00:15 +08:00
Frank Lee	f19b49e164	[booster] init module structure and definition (#3056)	2023-03-09 11:27:46 +08:00