dense(A)*sparse(B)

Author: yuehhua

Manifest.toml

@@ -142,10 +142,10 @@ deps = ["Libdl"]
 uuid = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 
 [[LogExpFunctions]]
-deps = ["DocStringExtensions", "IrrationalConstants", "LinearAlgebra"]
-git-tree-sha1 = "3d682c07e6dd250ed082f883dc88aee7996bf2cc"
+deps = ["ChainRulesCore", "DocStringExtensions", "IrrationalConstants", "LinearAlgebra"]
+git-tree-sha1 = "1f5097e3bce576e1cdf6dc9f051ab8c6e196b29e"
 uuid = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
-version = "0.3.0"
+version = "0.3.1"
 
 [[Logging]]
 uuid = "56ddb016-857b-54e1-b83d-db4d58db5568"

lib/cusparse/interfaces.jl

@@ -79,17 +79,22 @@ for (taga, untaga) in tag_wrappers, (wrapa, transa, unwrapa) in op_wrappers
     end
 end
 
+function LinearAlgebra.:(*)(A::Union{CuMatrix, CuSparseMatrix}, B::CuSparseMatrix{T}) where {T}
+    return mul!(similar(CuMatrix{T}, (size(A,1), size(B,2))), A, CuArray(B))
+end
+
+function LinearAlgebra.mul!(C::CuMatrix{T}, A::CuMatrix{T}, B::CuSparseMatrix{T}) where {T}
+    mul!(C, B', A', one(T), zero(T))
+    return C'
+end
+
 Base.:(+)(A::CuSparseMatrixCSR, B::CuSparseMatrixCSR) = geam(one(eltype(A)), A, one(eltype(A)), B, 'O')
 Base.:(-)(A::CuSparseMatrixCSR, B::CuSparseMatrixCSR) = geam(one(eltype(A)), A, -one(eltype(A)), B, 'O')
 
-Base.:(+)(A::CuSparseMatrixCSR, B::Adjoint{T,<:CuSparseMatrixCSR}) where {T} =
-    A + Transpose(conj(B.parent))
-Base.:(-)(A::CuSparseMatrixCSR, B::Adjoint{T,<:CuSparseMatrixCSR}) where {T} =
-    A - Transpose(conj(B.parent))
-Base.:(+)(A::Adjoint{T,<:CuSparseMatrixCSR}, B::CuSparseMatrixCSR) where {T} =
-    Transpose(conj(A.parent)) + B
-Base.:(-)(A::Adjoint{T,<:CuSparseMatrixCSR}, B::CuSparseMatrixCSR) where {T} =
-    Transpose(conj(A.parent)) - B
+Base.:(+)(A::CuSparseMatrixCSR, B::Adjoint{T,<:CuSparseMatrixCSR}) where {T} = A + Transpose(conj(B.parent))
+Base.:(-)(A::CuSparseMatrixCSR, B::Adjoint{T,<:CuSparseMatrixCSR}) where {T} = A - Transpose(conj(B.parent))
+Base.:(+)(A::Adjoint{T,<:CuSparseMatrixCSR}, B::CuSparseMatrixCSR) where {T} = Transpose(conj(A.parent)) + B
+Base.:(-)(A::Adjoint{T,<:CuSparseMatrixCSR}, B::CuSparseMatrixCSR) where {T} = Transpose(conj(A.parent)) - B
 Base.:(+)(A::Adjoint{T,<:CuSparseMatrixCSR}, B::Adjoint{T,<:CuSparseMatrixCSR}) where {T} =
     Transpose(conj(A.parent)) + B
 Base.:(-)(A::Adjoint{T,<:CuSparseMatrixCSR}, B::Adjoint{T,<:CuSparseMatrixCSR}) where {T} =

test/cusparse/interfaces.jl

@@ -137,6 +137,30 @@ using LinearAlgebra, SparseArrays
         @test C ≈ collect(dC)
     end
 
+    @testset "dense(A)*sparse(B) $elty" for elty in [Float32, Float64, ComplexF32, ComplexF64]
+        n = 10
+        A = rand(elty, n, n)
+        B = sprand(elty, n, n, rand())
+    
+        dA = CuArray(A)
+        dB = CUSPARSE.CuSparseMatrixCSR(B)
+    
+        C = A * B
+        dC = dA * dB
+        @test C ≈ collect(dC)
+        @test dC isa CuMatrix{elty}
+        
+        C = B * A
+        dC = dB * dA
+        @test C ≈ collect(dC)
+        @test dC isa CuMatrix{elty}
+    
+        C = B * B
+        dC = dB * dB
+        @test C ≈ collect(dC)
+        @test dC isa CuMatrix{elty}
+    end
+
     @testset "issue #1095 ($elty)" for elty in [Float32, Float64, ComplexF32, ComplexF64]
         # Test non-square matrices
         n, m, p = 10, 20, 4