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
..
utils	[gemini] accelerate inference (#3641)	2023-04-26 16:32:40 +08:00
__init__.py	[gemini] accelerate inference (#3641)	2023-04-26 16:32:40 +08:00
albert.py	[Gemini] add albert in test models. (#2075)	2022-12-05 14:09:34 +08:00
beit.py	[testing] add beit model for unit testings (#2196)	2022-12-26 17:35:36 +08:00
bert.py	[test] bert test in non-distributed way (#2074)	2022-12-05 13:32:16 +08:00
gpt2.py	[test] align model name with the file name. (#2045)	2022-11-30 15:45:26 +08:00
hanging_param_model.py	[test] align model name with the file name. (#2045)	2022-11-30 15:45:26 +08:00
inline_op_model.py	[testing] fix testing models (#2036)	2022-11-29 13:42:06 +08:00
nested_model.py	[testing] fix testing models (#2036)	2022-11-29 13:42:06 +08:00
registry.py	[test] refactored testing components (#324)	2022-03-11 15:50:28 +08:00
repeated_computed_layers.py	[Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127)	2022-12-13 16:34:10 +08:00
resnet.py	[zero] new interface for ShardedOptimv2 (#406)	2022-03-14 20:48:41 +08:00
simple_net.py	[testing] fix testing models (#2036)	2022-11-29 13:42:06 +08:00