ColossalAI/tests
Jiarui Fang 2938edf446
[Gemini] update the non model data record method in runtime memory tracer (#2128)
2022-12-13 17:11:31 +08:00
..
components_to_test [Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127) 2022-12-13 16:34:10 +08:00
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 [autoparallel] Add alpha beta (#1973) 2022-11-17 16:01:14 +08:00
test_engine
test_fx [Pipeline] Add Topo Class (#2059) 2022-12-02 18:13:20 +08:00
test_gemini [Gemini] update the non model data record method in runtime memory tracer (#2128) 2022-12-13 17:11:31 +08:00
test_layers
test_moe [test] align model name with the file name. (#2045) 2022-11-30 15:45:26 +08:00
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