Add batching rule for "is_complex", "conj"#44649
Add batching rule for "is_complex", "conj"#44649zou3519 wants to merge 5 commits intogh/zou3519/305/basefrom
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.
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
Summary: Pull Request resolved: pytorch#44649 To unblock pytorch#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