Frank Lee 9ec401a722 [autoparallel] added new strategy constructor template (#1661) ...

* [autoparallel] added new strategy constructor template

* polish code

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

..

__init__.py

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

2022-09-28 14:01:36 +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] protect bcast handler from invalid strategies (#1631)

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

conv_handler_v2.py

[autoparallel] add conv handler v2 (#1663)

2022-09-28 11:24:59 +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 node handler for bmm (#1655)

2022-09-28 11:32:16 +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

layer_norm_handler.py

[autoparallel] add layernorm handler (#1629)

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

node_handler.py

[autoparallel] add conv handler v2 (#1663)

2022-09-28 11:24:59 +08:00

operator_handler.py

[autoparallel] adapt solver with gpt (#1653)

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

registry.py

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

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

reshape_handler.py

[autoparallel] adapt solver with gpt (#1653)

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

strategy_generator.py

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

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

unary_elementwise_handler.py

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

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

where_handler.py

[autoparallel] where handler (#1651)

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