Frank Lee d925122020 [autoparallel] added new linear module handler (#1616)		2022-09-21 12:23:21 +08:00
..
op_handler	[autoparallel] added new linear module handler (#1616)	2022-09-21 12:23:21 +08:00
__init__.py	[autoparallel] added solver option dataclass (#1588)	2022-09-13 14:47:09 +08:00
_utils.py	[autoparallel] add bcast op handler (#1600)	2022-09-16 11:33:01 +08:00
constants.py	[autoparallel]add bcast matmul strategies (#1605)	2022-09-20 11:26:21 +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	[autoparallel] added new linear module handler (#1616)	2022-09-21 12:23:21 +08:00
solver.py	[autoparallel] adapt solver with resnet (#1583)	2022-09-13 12:07:09 +08:00
strategies_constructor.py	[fx] PoC of runtime shape consistency application (#1607)	2022-09-20 14:00:04 +08:00