ColossalAI/tests/test_auto_parallel
YuliangLiu0306 3345c6d352
[autoparellel]add strategies constructor (#1505)
* [autoparellel]add strategies constructor

* remove duplicated strategies

* polish code

* adapt cost graph with StrategiesConstructor

* polish
2022-08-30 16:32:09 +08:00
..
test_conv_handler.py [autoparallel] add more sharding strategies to conv (#1487) 2022-08-26 11:17:56 +08:00
test_cost_graph.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00
test_dot_handler.py [autoparallel] add cost graph class (#1481) 2022-08-25 17:19:59 +08:00
test_liveness_analysis.py [autoparallel] added liveness analysis (#1516) 2022-08-30 15:54:37 +08:00
test_strategies_constructor.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00