-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Add batching rule for "is_complex", "conj" #44649
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
Conversation
To unblock #43208, which adds "is_complex" checks to backward formulas that are being tested for batched gradient support with vmap. Test Plan: - `pytest test/test_vmap.py -v` [ghstack-poisoned]
To unblock #43208, which adds "is_complex" checks to backward formulas that are being tested for batched gradient support with vmap. Test Plan: - `pytest test/test_vmap.py -v` Differential Revision: [D23685356](https://our.internmc.facebook.com/intern/diff/D23685356) [ghstack-poisoned]
anjali411
left a comment
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.
Looks like test_vmap is not added in this list https://github.com/pytorch/pytorch/blob/master/test/run_test.py and that's why (despite the test failures), the CI was green for #43208
Thanks for pointing that out. I've added it to run_test in #44656. Fingers crossed that there are no Windows or rocm issues... |
To unblock #43208, which adds "is_complex" checks to backward formulas that are being tested for batched gradient support with vmap. Test Plan: - `pytest test/test_vmap.py -v` Differential Revision: [D23685356](https://our.internmc.facebook.com/intern/diff/D23685356) [ghstack-poisoned]
💊 CI failures summary and remediationsAs of commit b3442a4 (more details on the Dr. CI page):
1 job timed out:
ci.pytorch.org: 1 failedThis 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 on the GitHub issue tracker or post in the (internal) Dr. CI Users group. This comment has been revised 7 times. |
Codecov Report
@@ Coverage Diff @@
## gh/zou3519/305/base #44649 +/- ##
======================================================
Coverage ? 68.08%
======================================================
Files ? 384
Lines ? 49763
Branches ? 0
======================================================
Hits ? 33880
Misses ? 15883
Partials ? 0 Continue to review full report at Codecov.
|
To unblock #43208, which adds "is_complex" checks to backward formulas that are being tested for batched gradient support with vmap. Test Plan: - `pytest test/test_vmap.py -v` Differential Revision: [D23685356](https://our.internmc.facebook.com/intern/diff/D23685356) [ghstack-poisoned]
To unblock #43208, which adds "is_complex" checks to backward formulas that are being tested for batched gradient support with vmap. Test Plan: - `pytest test/test_vmap.py -v` Differential Revision: [D23685356](https://our.internmc.facebook.com/intern/diff/D23685356) [ghstack-poisoned]
Summary: Pull Request resolved: #44649 To unblock #43208, which adds "is_complex" checks to backward formulas that are being tested for batched gradient support with vmap. Test Plan: - `pytest test/test_vmap.py -v` Reviewed By: anjali411 Differential Revision: D23685356 Pulled By: zou3519 fbshipit-source-id: 29e41a9296336f6d1008e3040cade4c643bf5ebf
Stack from ghstack:
To unblock #43208, which adds "is_complex" checks to backward formulas
that are being tested for batched gradient support with vmap.
Test Plan:
pytest test/test_vmap.py -vDifferential Revision: D23685356