YuliangLiu0306 81f7530ee7 [autoparallel] adapt solver and CostGraph with new handler (#1695) ...

* [autoparallel] adapt solver and CostGraph with new handler

* fix test issue

2022-10-13 14:04:15 +08:00

..

__init__.py

[autoparallel] adapt solver and CostGraph with new handler (#1695)

2022-10-13 14:04:15 +08:00

batch_norm_handler_v2.py

[autoparallel] add batch norm handler v2 (#1666)

2022-09-29 11:02:49 +08:00

batch_norm_handler.py

[autoparallel] add bcast op handler (#1600)

2022-09-16 11:33:01 +08:00

bcast_op_handler.py

[autoparallel] update CommSpec (#1667)

2022-09-29 11:20:59 +08:00

broadcast.py

[autoparallel] added utils for broadcast operation (#1665)

2022-09-29 11:22:29 +08:00

conv_handler_v2.py

[autoparallel] adapt solver and CostGraph with new handler (#1695)

2022-10-13 14:04:15 +08:00

conv_handler.py

[autoparallel] add bcast op handler (#1600)

2022-09-16 11:33:01 +08:00

dot_handler_v2.py

[autoparallel] added sharding spec conversion for linear handler (#1687)

2022-10-12 11:16:18 +08:00

dot_handler.py

[autoparallel] adapt solver with gpt (#1653)

2022-09-28 11:17:26 +08:00

embedding_handler.py

[autoparallel] add embedding handler (#1620)

2022-09-23 12:34:30 +08:00

getitem_handler.py

[autoparallel] add following node generator (#1673)

2022-10-09 14:49:18 +08:00

layer_norm_handler_v2.py

[autoparallel] add layer norm handler v2 (#1671)

2022-10-09 14:23:22 +08:00

layer_norm_handler.py

[autoparallel] add layernorm handler (#1629)

2022-09-23 12:00:25 +08:00

node_handler.py

[autoparallel] add output handler and placeholder handler (#1694)

2022-10-13 13:42:36 +08:00

normal_pooling_handler.py

[autoparallel] add pooling handler (#1690)

2022-10-13 13:42:13 +08:00

operator_handler.py

[autoparallel] adapt solver with gpt (#1653)

2022-09-28 11:17:26 +08:00

output_handler.py

[autoparallel] add output handler and placeholder handler (#1694)

2022-10-13 13:42:36 +08:00

placeholder_handler.py

[autoparallel] add output handler and placeholder handler (#1694)

2022-10-13 13:42:36 +08:00

registry.py

[autoparallel] added new strategy constructor template (#1661)

2022-09-28 14:01:36 +08:00

reshape_handler_v2.py

[autoparallel] adapt solver and CostGraph with new handler (#1695)

2022-10-13 14:04:15 +08:00

reshape_handler.py

[hotfix] solver bug caused by dict type comm cost (#1686)

2022-10-11 17:57:03 +08:00

strategy_generator.py

[autoparallel] added all non-bcast matmul strategies (#1603)

2022-09-16 10:47:32 +08:00

unary_elementwise_handler_v2.py

[autoparallel] adapt solver and CostGraph with new handler (#1695)

2022-10-13 14:04:15 +08:00

unary_elementwise_handler.py

[autoparallel] change the following nodes strategies generation logic (#1636)

2022-09-27 11:20:52 +08:00

utils.py

[autoparallel] added sharding spec conversion for linear handler (#1687)

2022-10-12 11:16:18 +08:00

where_handler_v2.py

[autoparallel] where_handler_v2 (#1688)

2022-10-13 11:02:22 +08:00

where_handler.py

[autoparallel] where handler (#1651)

2022-09-27 11:20:43 +08:00