ColossalAI/tests
Jiarui Fang 27327a4c90
[example] add palm pytorch version (#2172)
2022-12-22 10:15:34 +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 [hotfix] fix aten default bug (#2158) 2022-12-20 22:40:46 +08:00
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel
test_ddp
test_device
test_engine
test_fx [Pipeline] Add Topo Class (#2059) 2022-12-02 18:13:20 +08:00
test_gemini [example] add palm pytorch version (#2172) 2022-12-22 10:15:34 +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 updated attention kernel (#2133) 2022-12-16 10:54:03 +08:00
test_zero [hotfix] fix auto policy of test_sharded_optim_v2 (#2157) 2022-12-20 23:03:18 +08:00
__init__.py