ColossalAI/tests
Frank Lee 73d3e4d309
[booster] implemented the torch ddd + resnet example (#3232)
* [booster] implemented the torch ddd + resnet example

* polish code
2023-03-27 10:24:14 +08:00
..
components_to_test
kit [FX] refactor experimental tracer and adapt it with hf models (#3157) 2023-03-22 10:40:33 +08:00
test_amp [workflow] only report coverage for changed files (#2524) 2023-01-30 21:28:27 +08:00
test_analyzer [Analyzer] fix analyzer tests (#3197) 2023-03-22 13:38:11 +08:00
test_auto_parallel [auto-parallel] add auto-offload feature (#3154) 2023-03-21 14:17:41 +08:00
test_autochunk [autochunk] support complete benchmark (#3121) 2023-03-13 17:42:37 +08:00
test_booster [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
test_checkpoint_io [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
test_cluster [API] implement device mesh manager (#3221) 2023-03-24 13:39:12 +08:00
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel
test_ddp
test_device
test_engine
test_fx [hotfix] skip torchaudio tracing test (#3211) 2023-03-24 12:15:33 +08:00
test_gemini [test] fixed torchrec model test (#3167) 2023-03-20 11:40:25 +08:00
test_layers
test_moe
test_ops
test_optimizer
test_pipeline
test_tensor [DTensor] refactor dtensor with new components (#3089) 2023-03-14 16:25:47 +08:00
test_trainer
test_utils [hotfix] skip torchaudio tracing test (#3211) 2023-03-24 12:15:33 +08:00
test_zero [zero] add zero wrappers (#2523) 2023-01-29 17:52:58 +08:00
__init__.py