ColossalAI/tests/test_booster
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
..
test_mixed_precision [booster] added the plugin base and torch ddp plugin (#3180) 2023-03-21 17:39:30 +08:00
test_plugin [booster] implement Gemini plugin (#3352) 2023-03-31 16:06:13 +08:00
test_accelerator.py [booster] added the plugin base and torch ddp plugin (#3180) 2023-03-21 17:39:30 +08:00