[AutoParallel] Add paddle.distributed.reshard python API. #57293
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
LiYuRio
reviewed
Sep 14, 2023
LiYuRio
reviewed
Sep 14, 2023
chenwhql
reviewed
Sep 14, 2023
|
中文Doc文档：PaddlePaddle/docs#6192 |
chenwhql
reviewed
Sep 18, 2023
| @@ -67,6 +67,7 @@ | |||
| from .auto_parallel import shard_op # noqa: F401 | |||
| from .auto_parallel.api import shard_tensor # noqa: F401 | |||
| from .auto_parallel.api import dtensor_from_fn # noqa: F401 | |||
| from .auto_parallel.api import reshard # noqa: F401 | |||
There was a problem hiding this comment.
这里是不是还需要将reshard加到all list中
chenwhql
reviewed
Sep 18, 2023
test/auto_parallel/CMakeLists.txt
Outdated
| @@ -186,6 +186,8 @@ if(WITH_DISTRIBUTE AND WITH_GPU) | |||
| py_test_modules(test_engine_save_load MODULES test_engine_save_load) | |||
| py_test_modules(test_rule_based_tuner MODULES test_rule_based_tuner) | |||
| py_test_modules(test_dist_tensor MODULES test_dist_tensor) | |||
| py_test_modules(test_reshard_api MODULES test_reshard_api) | |||
There was a problem hiding this comment.
这个CMakeLists里面对不同性质单测做了划分，reshard是个多卡单测，看看是不是和其他reshard_x_to_x单测放到一起，最好也加个超时控制，多卡单测容易超时
XieYunshen
approved these changes
Sep 19, 2023
Frida-a
pushed a commit
to Frida-a/Paddle
that referenced
this pull request
Oct 14, 2023
…le#57293) * Add paddle.distributed.reshard API. It supports reshard for DistTensor. * Polish code with review comments. * Fix problem of in_dynamic_mode * Fix some problems according to review comments. * Set test_reshard_api as multi-cards testcase. And set its timeout.
danleifeng
pushed a commit
to danleifeng/Paddle
that referenced
this pull request
Nov 14, 2023
…le#57293) * Add paddle.distributed.reshard API. It supports reshard for DistTensor. * Polish code with review comments. * Fix problem of in_dynamic_mode * Fix some problems according to review comments. * Set test_reshard_api as multi-cards testcase. And set its timeout.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Others
PR changes
APIs
Description
Pcard-73145
Add
paddle.distributed.reshard(dist_tensor, dist_attr)python API. It supports reshard for DistTensor.