Skip to content

Commit e4ac81a

Browse files
kshyattgdalle
andauthored
Simplify sparse type mappings (#3076)
* Simplify sparse type mappings * Update lib/cusparse/src/array.jl Co-authored-by: Guillaume Dalle <22795598+gdalle@users.noreply.github.com> --------- Co-authored-by: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
1 parent 49ce6f6 commit e4ac81a

1 file changed

Lines changed: 2 additions & 10 deletions

File tree

lib/cusparse/src/array.jl

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -105,23 +105,15 @@ function CUDACore.unsafe_free!(xs::CuSparseMatrixCSR)
105105
return
106106
end
107107

108-
GPUArrays.sparse_array_type(sa::CuSparseMatrixCSC) = CuSparseMatrixCSC
109108
GPUArrays.sparse_array_type(::Type{<:CuSparseMatrixCSC}) = CuSparseMatrixCSC
110-
GPUArrays.sparse_array_type(sa::CuSparseMatrixCSR) = CuSparseMatrixCSR
111109
GPUArrays.sparse_array_type(::Type{<:CuSparseMatrixCSR}) = CuSparseMatrixCSR
112-
GPUArrays.sparse_array_type(sa::CuSparseVector) = CuSparseVector
113110
GPUArrays.sparse_array_type(::Type{<:CuSparseVector}) = CuSparseVector
114-
GPUArrays.dense_array_type(sa::CuSparseVector) = CuArray
115111
GPUArrays.dense_array_type(::Type{<:CuSparseVector}) = CuArray
116-
GPUArrays.dense_array_type(sa::CuSparseMatrixCSC) = CuArray
117112
GPUArrays.dense_array_type(::Type{<:CuSparseMatrixCSC}) = CuArray
118-
GPUArrays.dense_array_type(sa::CuSparseMatrixCSR) = CuArray
119113
GPUArrays.dense_array_type(::Type{<:CuSparseMatrixCSR}) = CuArray
120114

121-
GPUArrays.csc_type(sa::CuSparseMatrixCSR) = CuSparseMatrixCSC
122-
GPUArrays.csr_type(sa::CuSparseMatrixCSC) = CuSparseMatrixCSR
123-
GPUArrays.coo_type(sa::Union{CuSparseMatrixCSR, Transpose{<:Any,<:CuSparseMatrixCSR}, Adjoint{<:Any,<:CuSparseMatrixCSR}}) = CuSparseMatrixCOO
124-
GPUArrays.coo_type(sa::Union{CuSparseMatrixCSC, Transpose{<:Any,<:CuSparseMatrixCSC}, Adjoint{<:Any,<:CuSparseMatrixCSC}}) = CuSparseMatrixCOO
115+
GPUArrays.csc_type(::Type{<:CuSparseMatrixCSR}) = CuSparseMatrixCSC
116+
GPUArrays.csr_type(::Type{<:CuSparseMatrixCSC}) = CuSparseMatrixCSR
125117
GPUArrays.coo_type(::Type{T}) where {T<:Union{CuSparseMatrixCSR, Transpose{<:Any,<:CuSparseMatrixCSR}, Adjoint{<:Any,<:CuSparseMatrixCSR}}} = CuSparseMatrixCOO
126118
GPUArrays.coo_type(::Type{T}) where {T<:Union{CuSparseMatrixCSC, Transpose{<:Any,<:CuSparseMatrixCSC}, Adjoint{<:Any,<:CuSparseMatrixCSC}}} = CuSparseMatrixCOO
127119

0 commit comments

Comments
 (0)