[Auto Parallel] Add the support for the auto completion of while_op #39939
Conversation
This file contains hidden or 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
phlrain previously approved these changes Mar 14, 2022
| import numpy as np | ||
| import paddle.nn as nn | ||
| import paddle.utils as utils | ||
| import paddle.fluid as fluid |
Contributor
There was a problem hiding this comment.
外部开发者不推荐使用fluid下的api，建议单测跟外部使用方式一致
| | ||
| input_array = fluid.layers.array_write(pred, i) | ||
| cond = fluid.layers.less_than(x=i, y=loop_len) | ||
| while_op = fluid.layers.While(cond=cond) |
Contributor
There was a problem hiding this comment.
这里是否可以使用paddle.static.nn.while_loop？
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/static/nn/while_loop_cn.html#while-loop
liqitong-a pushed a commit to liqitong-a/Paddle that referenced this pull request Mar 17, 2022
…addlePaddle#39939) * [Auto Parallel] Support the auto completion of while_op * [Auto Parallel] Improve the completion algorithms * [Auto Parallel] Fix bugs for ernie inference * [Auto Parallel] Remove attrs which cannot be pickled * [Auto Parallel] make the dims_mappings of LodTensorArray vars empty * [Auto Parallel] Fix bugs for the ernie inference in the pipeline parallel * [Auto Parallel] Remove unncessary comments * [Auto Parallel] Fix a bug of the CMakeLists * [Auto Parallel] Use the newest APIs to write the unit test * [Auto Parallel] Remove unnecessary statements
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
3 participants
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
New features
PR changes
Others
Describe