Add storage_type for Sparse CuArrays

JuliaSmoothOptimizers · Oct 9, 2024 · eb5d957 · eb5d957
1 parent d1b2c6c
commit eb5d957
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ext/LinearOperatorsCUDAExt.jl b/ext/LinearOperatorsCUDAExt.jl
@@ -1,8 +1,9 @@
 module LinearOperatorsCUDAExt
 
 using LinearOperators
-isdefined(Base, :get_extension) ? (using CUDA) : (using ..CUDA)
+isdefined(Base, :get_extension) ? (using CUDA; using CUDA.CUSPARSE) : (using ..CUDA; using ..CUDA.CUSPARSE)
 
 LinearOperators.storage_type(::CuArray{T, 2, D}) where {T, D} = CuArray{T, 1, D}
+LinearOperators.storage_type(::AbstractCuSparseMatrix{T}) where {T} = CuArray{T, 1, CUDA.DeviceMemory}
 
 end # module