ColossalAI/tests/test_zero
Hongxin Liu 50793b35f4
[gemini] accelerate inference (#3641)
* [gemini] support don't scatter after inference

* [chat] update colossalai strategy

* [chat] fix opt benchmark

* [chat] update opt benchmark

* [gemini] optimize inference

* [test] add gemini inference test

* [chat] fix unit test ci

* [chat] fix ci

* [chat] fix ci

* [chat] skip checkpoint test
2023-04-26 16:32:40 +08:00
..
test_gemini [gemini] accelerate inference (#3641) 2023-04-26 16:32:40 +08:00
test_legacy [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_low_level [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00