ColossalAI/colossalai/auto_parallel/solver
YuliangLiu0306 0908d0fc61
[autoparallel]add backward cost info into strategies (#1524)
2022-09-07 11:19:00 +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 [autoparallel]add backward cost info into strategies (#1524) 2022-09-07 11:19:00 +08:00
cost_graph.py [autoparallel] change the merge node logic (#1533) 2022-09-07 11:18:19 +08:00
dot_handler.py
graph_analysis.py [autoparallel] added liveness analysis (#1516) 2022-08-30 15:54:37 +08:00
operator_handler.py [autoparallel]add backward cost info into strategies (#1524) 2022-09-07 11:19:00 +08:00
sharding_strategy.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00
strategies_constructor.py [autoparallel] support fucntion in operator handler (#1529) 2022-09-07 11:18:41 +08:00