-
Notifications
You must be signed in to change notification settings - Fork 457
Fix batch computation in Pivoted Cholesky #2823
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
|
This pull request was exported from Phabricator. Differential Revision: D72906531 |
|
This pull request was exported from Phabricator. Differential Revision: D72906531 |
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #2823 +/- ##
=========================================
Coverage 100.00% 100.00%
=========================================
Files 211 211
Lines 19286 19286
=========================================
Hits 19286 19286 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
Summary: ## Context Resolves issue meta-pytorch#2819 where `PivotedCholesky.update_` break when there is more than a single batch dimension. ## Changes Updates a line to extend boolean indexing logic to cases where `len(batch_shape) > 1` Reviewed By: saitcakmak Differential Revision: D72906531
|
This pull request was exported from Phabricator. Differential Revision: D72906531 |
Summary: ## Context Resolves issue meta-pytorch#2819 where `PivotedCholesky.update_` break when there is more than a single batch dimension. ## Changes Updates a line to extend boolean indexing logic to cases where `len(batch_shape) > 1` Reviewed By: saitcakmak Differential Revision: D72906531
|
This pull request was exported from Phabricator. Differential Revision: D72906531 |
Summary: ## Context Resolves issue meta-pytorch#2819 where `PivotedCholesky.update_` break when there is more than a single batch dimension. ## Changes Updates a line to extend boolean indexing logic to cases where `len(batch_shape) > 1` Reviewed By: saitcakmak Differential Revision: D72906531
Summary: Pull Request resolved: meta-pytorch#2823 ## Context Resolves issue meta-pytorch#2819 where `PivotedCholesky.update_` break when there is more than a single batch dimension. ## Changes Updates a line to extend boolean indexing logic to cases where `len(batch_shape) > 1` Reviewed By: saitcakmak Differential Revision: D72906531
|
This pull request was exported from Phabricator. Differential Revision: D72906531 |
|
This pull request has been merged in cbada4c. |
|
Closes #2819 |
Summary:
Summary
TODO:
Changes
Updates a line containing indexing logic that breaks when
len(batch_shape) > 1Differential Revision: D72906531