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] Fix crashes during Conv backward pass #113398
Conversation
By adding weights tensor to the MPSGraph cache key. Add regression test to validate that collision no longer happens Fixes #112998
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/113398
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (1 Unrelated Failure)As of commit 619fa1d with merge base 92e7f79 (): UNSTABLE - The following job failed but was likely due to flakiness present on trunk and has been marked as unstable:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
@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). Please use Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
By adding weights tensor to the MPSGraph cache key. Add regression test to validate that collision no longer happens Fixes pytorch#112998 Pull Request resolved: pytorch#113398 Approved by: https://github.com/kulinseth
By adding weights tensor to the MPSGraph cache key.
Add regression test to validate that collision no longer happens
Fixes #112998