Have convert call CuSparseArray's constructors.

JuliaGPU · Sep 28, 2020 · b32cff9 · b32cff9
1 parent b529985
commit b32cff9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/cusparse/array.jl b/lib/cusparse/array.jl
@@ -12,6 +12,8 @@ abstract type AbstractCuSparseArray{Tv, N} <: AbstractSparseArray{Tv, Cint, N} e
 const AbstractCuSparseVector{Tv} = AbstractCuSparseArray{Tv,1}
 const AbstractCuSparseMatrix{Tv} = AbstractCuSparseArray{Tv,2}
 
+Base.convert(T::Type{<:AbstractCuSparseArray}, m::AbstractArray) = m isa T ? m : T(m)
+
 mutable struct CuSparseVector{Tv} <: AbstractCuSparseVector{Tv}
  iPtr::CuVector{Cint}
  nzVal::CuVector{Tv}