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
[pt2] add meta for linalg_ldl_solve
#101367
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/101367
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 777e564: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
ghstack-source-id: 1648d8fe93f0f6f48932540f08ab7a3a6073df59 Pull Request resolved: #101367
Part of this stack because it unlocks tests next to the prev PR in the stack. |
device=B.device, | ||
) | ||
|
||
|
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.
See TORCH_META_FUNC(linalg_ldl_solve)
in core.
check( | ||
LD.dtype == B.dtype, | ||
lambda: f"torch.linalg.ldl_solve: LD dtype {LD.dtype} does not match b dtype {B.dtype}", | ||
) |
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.
linearSolveCheckInputs
already has a dtype check
f"torch.linalg.ldl_solve: Expected B to have at least 2 dimensions, " | ||
f"but it has {B.ndim} dimensions instead" | ||
), | ||
) |
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.
This ndim
check will never be reached because self.size(-2)
is already checked in linearSolveCheckInputs
Pull Request resolved: #101367 Approved by: https://github.com/lezcano
Stack from ghstack:
linalg_lu_factor_ex
#101375SymInt
support forlinalg_lu
#101372linalg_ldl_solve
#101367linalg_ldl_factor_ex
#101362SymInt
support forlinalg_householder_product
#101315