-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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 torch frontend blas_and_lapack_ops cholesky #28767
Conversation
locally, for the 5 frontend tests of test_torch_cholesky, only one of paddle fails with RuntimeError('(NotFound) The kernel with key (CPU, Undefined(AnyLayout), bfloat16) of kernel |
Now it seems like the target test_torch_cholesky is no longer failing according to the github pipeline |
…e it should not be casted to float64
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.
Thanks for the contribution @Daniel4078!
I've just had a look at the failing test for torch.linalg.cholesky
, and it seems the problem is actually with the test itself. I was able to fix it for all backends by changing this line:
x = np.matmul(x.T, x) + np.identity(x.shape[0])
to this:
x = np.matmul(np.conjugate(x.T), x) + np.identity(x.shape[0], dtype=dtype[0])
Would you be able to make this change, and see if it also works for the version in blas_and_lapack_ops? (you can remove all these current changes you made) Thanks!
…r one in test_linalg to fix the problem
@Sam-Armstrong Thank you! I have checked, your changes also work in this blas_and_lapack_ops version. And I think this PR should be ready to be merged. Just in case, have you already pushed the fix to the linalg version or should I include it in this PR? |
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 good to me. Thanks a lot @Daniel4078!
PR Description
Related Issue
Closes #28766
Checklist
Socials