specialized _sparse similar methods

Author: Fe-r-oz

ext/QuantumToolboxCUDAExt.jl

@@ -106,13 +106,10 @@ QuantumToolbox.to_dense(::Type{T}, A::AbstractCuSparseArray) where {T<:Number} =
 
 QuantumToolbox._sparse_similar(A::CuSparseMatrixCSC, args...) = sparse(args..., fmt = :csc)
 QuantumToolbox._sparse_similar(A::CuSparseMatrixCSR, args...) = sparse(args..., fmt = :csr)
-function QuantumToolbox._sparse_similar(A::CuSparseMatrixCSC, rows, cols, vals, m::Int64, n::Int64)
-    cpu_sparse = sparse(rows, cols, vals, m, n)
-    return CuSparseMatrixCSC(cpu_sparse)
-end
-function QuantumToolbox._sparse_similar(A::CuSparseMatrixCSR, rows, cols, vals, m::Int64, n::Int64)
-    cpu_sparse = sparse(rows, cols, vals, m, n)
-    return CuSparseMatrixCSR(cpu_sparse)
-end
-
+_sparse_similar(A::CuSparseMatrixCSC, I::AbstractVector, J::AbstractVector, V::AbstractVector, m::Int, n::Int) =
+    CuSparseMatrixCSC(sparse(I, J, V, m, n))
+_sparse_similar(A::CuSparseMatrixCSC, m::Int, n::Int) = CuSparseMatrixCSC(spzeros(eltype(A), m, n))
+_sparse_similar(A::CuSparseMatrixCSR, I::AbstractVector, J::AbstractVector, V::AbstractVector, m::Int, n::Int) =
+    CuSparseMatrixCSR(sparse(I, J, V, m, n))
+_sparse_similar(A::CuSparseMatrixCSR, m::Int, n::Int) = CuSparseMatrixCSR(spzeros(eltype(A), m, n))
 end