ColossalAI/tests/test_booster
Frank Lee 638a07a7f9
[test] fixed gemini plugin test (#3411)
* [test] fixed gemini plugin test

* polish code

* polish code
2023-04-03 17:12:22 +08:00
..
test_mixed_precision [booster] added the plugin base and torch ddp plugin (#3180) 2023-03-21 17:39:30 +08:00
test_plugin [test] fixed gemini plugin test (#3411) 2023-04-03 17:12:22 +08:00
test_accelerator.py [booster] added the plugin base and torch ddp plugin (#3180) 2023-03-21 17:39:30 +08:00