github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-25 05:22:43 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
6ef70114628b83d39716e9746c062c980cd4a3be
ColossalAI/tests
History
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
..
components_to_test
[gemini] accelerate inference (#3641)
2023-04-26 16:32:40 +08:00
kit
[booster] add low level zero plugin (#3594)
2023-04-26 14:37:25 +08:00
test_amp
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_analyzer
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_auto_parallel
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_autochunk
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_booster
[booster] add low level zero plugin (#3594)
2023-04-26 14:37:25 +08:00
test_checkpoint_io
[checkpoint] support huggingface style sharded checkpoint (#3461)
2023-04-06 16:23:39 +08:00
test_cluster
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_comm
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_config
[pipeline] refactor the pipeline module (#1087)
2022-06-10 11:27:38 +08:00
test_context
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_data
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_data_pipeline_tensor_parallel
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_ddp
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_device
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_engine
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_fx
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_layers
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_moe
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_ops
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_optimizer
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_pipeline
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_tensor
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_trainer
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_utils
[test] refactor tests with spawn (#3452)
2023-04-06 14:51:35 +08:00
test_zero
[gemini] accelerate inference (#3641)
2023-04-26 16:32:40 +08:00
__init__.py
[zero] Update sharded model v2 using sharded param v2 (#323)
2022-03-11 15:50:28 +08:00