-
Notifications
You must be signed in to change notification settings - Fork 38
feat: more linear algebra operations (non factorization changes) #1883
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
src/TracedRArray.jl
Outdated
|
|
||
| # Base.first is very common usecase for getting first element to get the type | ||
| # inside LinearAlgebra.jl | ||
| Base.first(x::TracedRArray{T,N}) where {T,N} = @allowscalar(x[1]) |
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.
do we want to put the allowscalar here? I would've expected not (but we look to what gpuarrays do)
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.
let me remove this and check if anything breaks. I ended up wrapping more of the functionality manually (mostly because of conditionals)
2d4485f to
ba3251a
Compare
* refactor: move LU into a separate file * feat: lower cholesky * feat: lowering cholesky ldiv * test: cholesky * fix: revert change to is symm/hermitian
d6b5975 to
9b183f4
Compare
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
factorization ops are a bit messed up upstream, need to polish those before supporting them.
det/logabsdetinvcrossrmul!,lmul!,rdiv!,ldiv!-- array and numbernormalize/normalize!(via__normalize!)normissymmetric,ishermitian,isbanded(by extensionisdiag)choleskyNon Linear Algebera additions
nextfloatprevfloat