ColossalAI/tests/components_to_test/utils
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
..
__init__.py [gemini] accelerate inference (#3641) 2023-04-26 16:32:40 +08:00
dummy_data_generator.py [unit test] Refactored test cases with component func (#339) 2022-03-11 15:50:28 +08:00
executor.py [gemini] accelerate inference (#3641) 2023-04-26 16:32:40 +08:00