ColossalAI/colossalai/auto_parallel/solver
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
..
__init__.py [autoparallel] added liveness analysis (#1516) 2022-08-30 15:54:37 +08:00
constants.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00
conv_handler.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00
cost_graph.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00
dot_handler.py [autoparallel] added namespace constraints (#1490) 2022-08-24 15:44:07 +08:00
graph_analysis.py [autoparallel] added liveness analysis (#1516) 2022-08-30 15:54:37 +08:00
operator_handler.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00
sharding_strategy.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00
strategies_constructor.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00