YuliangLiu0306 21d6a48f4d [autoparallel] add shard option (#2696) ...

* [autoparallel] add shard option

* polish

2023-02-15 13:48:28 +08:00

..

test_gpt

[autoparallel] add shard option (#2696)

2023-02-15 13:48:28 +08:00

test_metainfo

[autoparallel] add shard option (#2696)

2023-02-15 13:48:28 +08:00

test_node_handler

[autoparallel] add shard option (#2696)

2023-02-15 13:48:28 +08:00

__init__.py

[autoparallel] handled illegal sharding strategy (#1728)

2022-10-19 12:53:06 +08:00

test_bias_addition_forward.py

[autoparallel] adapt autoparallel tests with latest api (#2626)

2023-02-08 15:02:12 +08:00

test_broadcast.py

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

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

test_checkpoint.py

[autoparallel] support origin activation ckpt on autoprallel system (#2468)

2023-01-16 16:25:13 +08:00

test_compatibility_with_ddp.py

[autoparallel] test compatibility for gemini and auto parallel (#2700)

2023-02-15 09:43:29 +08:00

test_compatibility_with_gemini.py

[autoparallel] test compatibility for gemini and auto parallel (#2700)

2023-02-15 09:43:29 +08:00

test_liveness_analysis.py

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

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

test_param_resharding_cost.py

[autoparallel] add shard option (#2696)

2023-02-15 13:48:28 +08:00

test_shape_consistency_pass.py

[autoparallel] adapt autoparallel tests with latest api (#2626)

2023-02-08 15:02:12 +08:00

test_solver_with_resnet_v2.py

[autoparallel] add shard option (#2696)

2023-02-15 13:48:28 +08:00