YuliangLiu0306 d164449d00 [autoparallel] add resnet autoparallel unit test and add backward weight communication cost (#1589)		2022-09-13 18:05:05 +08:00
..
op_handler	[autoparallel] add resnet autoparallel unit test and add backward weight communication cost (#1589)	2022-09-13 18:05:05 +08:00
__init__.py	[autoparallel] added solver option dataclass (#1588)	2022-09-13 14:47:09 +08:00
_utils.py	[autoparallel] added generate_sharding_spec to utils (#1590)	2022-09-13 15:43:22 +08:00
constants.py	[autoparallel] adapt solver with resnet (#1583)	2022-09-13 12:07:09 +08:00
cost_graph.py	[autoparallel] adapt solver with resnet (#1583)	2022-09-13 12:07:09 +08:00
graph_analysis.py	[autoparallel] adapt solver with resnet (#1583)	2022-09-13 12:07:09 +08:00
options.py	[autoparallel] added solver option dataclass (#1588)	2022-09-13 14:47:09 +08:00
sharding_strategy.py	[autoparellel]add strategies constructor (#1505)	2022-08-30 16:32:09 +08:00
solver.py	[autoparallel] adapt solver with resnet (#1583)	2022-09-13 12:07:09 +08:00
strategies_constructor.py	[autoparallel] added generate_sharding_spec to utils (#1590)	2022-09-13 15:43:22 +08:00