Skip to content

Commit efa8656

Browse files
committed
Simplify sparse type mappings
1 parent 2e540fd commit efa8656

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

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)