ColossalAI/tests/test_auto_parallel
YuliangLiu0306 ffcdbf0f65
[autoparallel]integrate auto parallel feature with new tracer (#3408)
* [autoparallel] integrate new analyzer in module level

* unify the profiling method

* polish

* fix no codegen bug

* fix pass bug

* fix liveness test

* polish
2023-04-04 17:40:45 +08:00
..
test_ckpt_solvers [hotfix] skip auto checkpointing tests (#3029) 2023-03-07 15:50:00 +08:00
test_offload [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
test_pass [autoparallel]integrate auto parallel feature with new tracer (#3408) 2023-04-04 17:40:45 +08:00
test_tensor_shard [autoparallel]integrate auto parallel feature with new tracer (#3408) 2023-04-04 17:40:45 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00