Added few SparseArrays functions (#2254)

Author: albertomercurio

lib/cusparse/array.jl

@@ -301,6 +301,26 @@ SparseArrays.nonzeros(g::AbstractCuSparseArray) = g.nzVal
 SparseArrays.nonzeroinds(g::AbstractCuSparseVector) = g.iPtr
 
 SparseArrays.rowvals(g::CuSparseMatrixCSC) = g.rowVal
+SparseArrays.getcolptr(S::CuSparseMatrixCSC) = S.colPtr
+
+function SparseArrays.findnz(S::MT) where {MT <: AbstractCuSparseMatrix}
+    S2 = CuSparseMatrixCOO(S)
+    I = S2.rowInd
+    J = S2.colInd
+    V = S2.nzVal
+
+    # To make it compatible with the SparseArrays.jl version
+    idxs = sortperm(J)
+    I = I[idxs]
+    J = J[idxs]
+    V = V[idxs]
+
+    return (I, J, V)
+end
+
+function SparseArrays.sparsevec(I::CuArray{Ti}, V::CuArray{Tv}, n::Integer) where {Ti,Tv}
+    CuSparseVector(I, V, n)
+end
 
 LinearAlgebra.issymmetric(M::Union{CuSparseMatrixCSC,CuSparseMatrixCSR}) = size(M, 1) == size(M, 2) ? norm(M - transpose(M), Inf) == 0 : false
 LinearAlgebra.ishermitian(M::Union{CuSparseMatrixCSC,CuSparseMatrixCSR}) = size(M, 1) == size(M, 2) ? norm(M - adjoint(M), Inf) == 0 : false

test/libraries/cusparse.jl

@@ -2,7 +2,7 @@ using CUDA.CUSPARSE
 
 using LinearAlgebra
 using SparseArrays
-using SparseArrays: nonzeroinds
+using SparseArrays: nonzeroinds, getcolptr
 
 @test CUSPARSE.version() isa VersionNumber
 
@@ -106,6 +106,22 @@ blockdim = 5
     d_x = LowerTriangular(CuSparseMatrixCSC(x))
     @test !istriu(d_x)
     @test istril(d_x)
+
+    A = sprand(n, n, 0.2)
+    d_A = CuSparseMatrixCSC(A)
+    @test Array(getcolptr(d_A)) == getcolptr(A)
+    i, j, v = findnz(A)
+    d_i, d_j, d_v = findnz(d_A)
+    @test Array(d_i) == i && Array(d_j) == j && Array(d_v) == v
+    i = unique(sort(rand(1:n, 10)))
+    vals = rand(length(i))
+    d_i = CuArray(i)
+    d_vals = CuArray(vals)
+    v = sparsevec(i, vals, n)
+    d_v = sparsevec(d_i, d_vals, n)
+    @test Array(d_v.iPtr) == v.nzind
+    @test Array(d_v.nzVal) == v.nzval
+    @test d_v.len == v.n
 end
 
 @testset "construction" begin