-
Notifications
You must be signed in to change notification settings - Fork 21.4k
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.
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
[MPS] Convolution cleanup; remove unnecessary contiguous calls #95078
Conversation
DenisVieriu97
commented
Feb 17, 2023
- Fixes convolution crashes in backward with weights
- Removes unnecessary contiguous calls
…ntiguous calls (#341) * Fix convolution crash; remove unnecessary contiguous calls * Fix lintrunner
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Lets hold off till we fix the is_view
part of the code and test : nn/test_convolution.py
Co-authored-by: Ramin Azarmehr <razarmehr@apple.com>
@pytorchbot merge -f "MPS tests are green." |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
…ch#95078) - Fixes convolution crashes in backward with weights - Removes unnecessary contiguous calls Pull Request resolved: pytorch#95078 Approved by: https://github.com/kulinseth
* [MPS] Convolution cleanup; remove unnecessary contiguous calls (#95078) - Fixes convolution crashes in backward with weights - Removes unnecessary contiguous calls Pull Request resolved: #95078 Approved by: https://github.com/kulinseth * [MPS] Fix nn.functional.conv_transpose2d grad (#94871) - add _mps_convolution_impl that takes optional shape - for conv_tranpose2d grad, use the shape from forward pass directly - for conv, calculate the shape from input - remove nn.functional.conv_transpose2d grad from blocklist Pull Request resolved: #94871 Approved by: https://github.com/kulinseth --------- Co-authored-by: Denis Vieriu <104024078+DenisVieriu97@users.noreply.github.com> Co-authored-by: Denis Vieriu <dvieriu@apple.com>
- Fixes convolution crashes in backward with weights - Removes unnecessary contiguous calls Pull Request resolved: pytorch/pytorch#95078 Approved by: https://github.com/kulinseth
pytorch#95078)" This reverts commit 5e47571.
* [MPS] Convolution cleanup; remove unnecessary contiguous calls (pytorch#95078) - Fixes convolution crashes in backward with weights - Removes unnecessary contiguous calls Pull Request resolved: pytorch#95078 Approved by: https://github.com/kulinseth * [MPS] Fix nn.functional.conv_transpose2d grad (pytorch#94871) - add _mps_convolution_impl that takes optional shape - for conv_tranpose2d grad, use the shape from forward pass directly - for conv, calculate the shape from input - remove nn.functional.conv_transpose2d grad from blocklist Pull Request resolved: pytorch#94871 Approved by: https://github.com/kulinseth --------- Co-authored-by: Denis Vieriu <104024078+DenisVieriu97@users.noreply.github.com> Co-authored-by: Denis Vieriu <dvieriu@apple.com>