YuliangLiu0306 0dbcd4a6f5 [autoparallel] add split handler (#2032) ...

* [autoparallel] add split handler

* add numerical test and runtime passes

2022-11-29 11:03:51 +08:00

..

experimental

[autoparallel] add split handler (#2032)

2022-11-29 11:03:51 +08:00

strategy

[autoparallel] add split handler (#2032)

2022-11-29 11:03:51 +08:00

__init__.py

[autoparallel] add experimental permute handler (#2029)

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

addmm_handler.py

[autoparallel] support addmm in tracer and solver (#1961)

2022-11-16 14:59:18 +08:00

batch_norm_handler.py

[autoparallel] resnet block runtime apply (#1709)

2022-10-17 13:37:38 +08:00

binary_elementwise_handler.py

[autoparallel]add essential CommActions for broadcast oprands (#1793)

2022-11-04 18:36:42 +08:00

bmm_handler.py

[autoparallel]add essential CommActions for broadcast oprands (#1793)

2022-11-04 18:36:42 +08:00

conv_handler.py

[autoparallel] fixed wrong sharding strategy in conv handler (#1747)

2022-10-20 16:12:39 +08:00

getatrr_handler.py

[autoparallel] add getattr handler (#1767)

2022-11-03 12:31:33 +08:00

getitem_handler.py

[autoparallel] refactored the autoparallel module for organization (#1706)

2022-10-14 13:27:00 +08:00

layer_norm_handler.py

[autoparallel] refactored the autoparallel module for organization (#1706)

2022-10-14 13:27:00 +08:00

linear_handler.py

[autoparallel] add runtime pass and numerical test for view handler (#2018)

2022-11-25 15:50:16 +08:00

matmul_handler.py

[autoparallel] support addmm in tracer and solver (#1961)

2022-11-16 14:59:18 +08:00

node_handler.py

[autoparallel] use pytree map style to process data (#1989)

2022-11-21 10:44:22 +08:00

normal_pooling_handler.py

[autoparallel] refactored the autoparallel module for organization (#1706)

2022-10-14 13:27:00 +08:00

output_handler.py

[autoparallel] support distributed dataloader option (#1906)

2022-11-17 20:11:53 +08:00

placeholder_handler.py

[autoparallel] support distributed dataloader option (#1906)

2022-11-17 20:11:53 +08:00

registry.py

[autoparallel] added binary elementwise node handler (#1758)

2022-10-25 14:32:01 +08:00

reshape_handler.py

[autoparallel] add split handler (#2032)

2022-11-29 11:03:51 +08:00

unary_elementwise_handler.py

[autoparallel] adapt handlers with attention block (#1990)

2022-11-21 10:44:11 +08:00

where_handler.py

[autoparallel] adapt handlers with attention block (#1990)

2022-11-21 10:44:11 +08:00