ColossalAI/tests
YuliangLiu0306 3b1b91eaf4
[autoparallel] record parameter attribute in colotracer (#2217)
* [autoparallel] record parameter attribute in collotracer

* [autoparallel] fix construct_meta_info bug
2022-12-28 19:29:08 +08:00
..
components_to_test [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_amp
test_auto_parallel [autoparallel] record parameter attribute in colotracer (#2217) 2022-12-28 19:29:08 +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 [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +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 [builder] unified cpu_optim fused_optim inferface (#2190) 2022-12-23 20:57:41 +08:00
test_pipeline [PP Middleware] Add bwd and step for PP middleware (#2111) 2022-12-12 12:40:03 +08:00
test_tensor [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_trainer
test_utils updated attention kernel (#2133) 2022-12-16 10:54:03 +08:00
test_zero [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
__init__.py