New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[autoparallel] shard param and buffer as expected #1753

Merged

FrankLeeeee merged 2 commits into hpcaitech:main from YuliangLiu0306:feature/update_runtime_pass

Oct 21, 2022

Contributor

YuliangLiu0306 commented Oct 20, 2022

what does this PR do

as PR([autoparallel] fixed shape consistency backward not invoked error #1745) mentioned, the previous implementation may lead to autograd backward function not invoked, this problem has been solved in this PR.
Update the solution_annotation_pass to make sure Parameters and Buffers are shard as expected.
Refactor the runtime apply function and remove the previous autograd function.


          [autoparallel] shard param and buffer as expected

79dd98c

YuliangLiu0306 added the Run Build and Test label


          fix unit test issue

1d652e2

FrankLeeeee approved these changes

View reviewed changes

FrankLeeeee merged commit 980ed21 into hpcaitech:main

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet