ColossalAI/tests
Jiarui Fang 5efda69735
[Gemini] hotfix the unittest bugs (#2125)
2022-12-13 14:14:55 +08:00
..
components_to_test
test_amp
test_auto_parallel [autoparallel] gpt2lp runtimee test (#2113) 2022-12-12 18:06:40 +08:00
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel
test_ddp
test_device
test_engine
test_fx
test_gemini [Gemini] hotfix the unittest bugs (#2125) 2022-12-13 14:14:55 +08:00
test_layers
test_moe
test_ops
test_optimizer [optimizer] add div_scale for optimizers (#2117) 2022-12-12 17:58:57 +08:00
test_pipeline [PP Middleware] Add bwd and step for PP middleware (#2111) 2022-12-12 12:40:03 +08:00
test_tensor [Gemini] remove static tracer (#2083) 2022-12-06 12:53:58 +08:00
test_trainer
test_utils
test_zero [Gemini] remove GLOBAL_MODEL_DATA_TRACER (#2091) 2022-12-06 22:30:16 +08:00
__init__.py