-
Notifications
You must be signed in to change notification settings - Fork 22.1k
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
Fix broadcasting of cdist backward #56605
Conversation
[ghstack-poisoned]
💊 CI failures summary and remediationsAs of commit 03ea991 (more details on the Dr. CI page):
This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.Please report bugs/suggestions to the (internal) Dr. CI Users group. |
hm, test_variant_consistency still fails with wrong size gradients. |
Yes I didn't run the cuda tests locally, so I was expecting something like that... |
Fix #55370 [ghstack-poisoned]
@ngimel updated this PR and the one above to fix also the CUDA implementation that was suffering from similar issues. |
Looks good, let's see what tests say. |
ci is good! :) |
Fix #55370 Differential Revision: [D27939202](https://our.internmc.facebook.com/intern/diff/D27939202) [ghstack-poisoned]
Fix #55370 Differential Revision: [D27939202](https://our.internmc.facebook.com/intern/diff/D27939202) [ghstack-poisoned]
Summary: Pull Request resolved: pytorch#56605 Fix pytorch#55370 Test Plan: Imported from OSS Reviewed By: ailzhang Differential Revision: D27939202 Pulled By: albanD fbshipit-source-id: a4ac50a7b504c24f47f5343414fb57523546a0c7
Fix #55370
Stack from ghstack:
Differential Revision: D27939202