ColossalAI/tests
Jiarui Fang 0aab52301e
[hotfix] fix a bug in model data stats tracing (#655)
2022-04-03 21:48:06 +08:00
..
components_to_test [zero] adapt zero for unsharded paramters (Optimizer part) (#601) 2022-04-01 20:10:47 +08:00
test_amp
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
test_engine
test_layers
test_moe [hotfix] fix a bug in model data stats tracing (#655) 2022-04-03 21:48:06 +08:00
test_optimizer
test_trainer [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
test_utils [hotfix] fix bugs in testing (#659) 2022-04-02 21:58:47 +08:00
test_zero_data_parallel [zero] adapt zero for unsharded paramters (Optimizer part) (#601) 2022-04-01 20:10:47 +08:00
__init__.py