Frank Lee 4973157ad7 [autoparallel] added sharding spec conversion for linear handler (#1687)

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

..

__init__.py

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

2022-10-09 14:23:22 +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] 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 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] added sharding spec conversion for linear handler (#1687)

2022-10-12 11:16:18 +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_v2.py

[autoparallel] add reshape handler v2 and fix some previous bug (#1683)

2022-10-11 18:12:59 +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] add unary element wise handler v2 (#1674)

2022-10-09 17:30:42 +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.py

[autoparallel] where handler (#1651)

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