Uh oh!
There was an error while loading. Please reload this page.
Fix sparse gemm, gemv, and mul for complex types - #536
Conversation
| (:onemklZsparse_set_csr_data_64, :ComplexF64, :Int64)) | ||
| @eval begin | ||
| function oneSparseMatrixCSR( |
There was a problem hiding this comment.
Do we also have a similar constructor for oneSparseMatrixCOO?
There was a problem hiding this comment.
Actually, I didn't want to add the constructor changes in this commit. I'll remove those.
| (:onemklZsparse_set_csr_data_64, :ComplexF64, :Int64)) | ||
| @eval begin | ||
| function oneSparseMatrixCSR( |
There was a problem hiding this comment.
Do we have a similar constructor for oneSparseMatrixCOO?
| # Matrix-vector multiplication with transpose/adjoint | ||
| function LinearAlgebra.generic_matvecmul!(C::oneVector{T}, tA::AbstractChar, A::Transpose{T, <:oneSparseMatrixCSR{T}}, B::oneVector{T}, _add::MulAddMul) where {T <: BlasFloat} | ||
| tA = tA in ('S', 's', 'H', 'h') ? 'N' : tA | ||
| tA_final = tA == 'N' ? 'T' : (tA == 'T' ? 'N' : 'C') |
There was a problem hiding this comment.
@michel2323 In the case tA = 'C', you need to do a product with the conjugate.
There was a problem hiding this comment.
I am wondering if you didn't mixed BlasReal with BlasFloat?
| function LinearAlgebra.generic_matvecmul!(C::oneVector{T}, tA::AbstractChar, A::Transpose{T, <:oneSparseMatrixCSC{T}}, B::oneVector{T}, _add::MulAddMul) where {T <: BlasFloat} | ||
| tA = tA in ('S', 's', 'H', 'h') ? 'N' : tA | ||
| tA_final = tA == 'N' ? 'T' : (tA == 'T' ? 'N' : 'C') |
There was a problem hiding this comment.
Similar issue, you need a special handle of tA = 'C'.
| function LinearAlgebra.generic_matvecmul!(C::oneVector{T}, tA::AbstractChar, A::Transpose{T, <:oneSparseMatrixCOO{T}}, B::oneVector{T}, _add::MulAddMul) where {T <: BlasFloat} | ||
| tA = tA in ('S', 's', 'H', 'h') ? 'N' : tA | ||
| tA_final = tA == 'N' ? 'T' : (tA == 'T' ? 'N' : 'C') |
There was a problem hiding this comment.
Once again, special handle of tA = 'C' is needed (when the matrix is complex).
Codecov Report❌ Patch coverage is
Additional details and impacted files@@ Coverage Diff @@## master #536 +/- ##
==========================================
- Coverage 79.20% 77.08% -2.13%
==========================================
Files 47 47 Lines 2996 3168 +172 ==========================================
+ Hits 2373 2442 +69 - Misses 623 726 +103 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
No description provided.