Add size for SparseMatrixCSR and SparseMatrixCOO

JuliaSparse · Nov 10, 2022 · 81c4201 · 81c4201
1 parent f9fbbd3
commit 81c4201
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/mklsparsematrix.jl b/src/mklsparsematrix.jl
@@ -18,6 +18,9 @@ mutable struct SparseMatrixCSR{Tv,Ti} <: AbstractSparseMatrix{Tv,Ti}
   nzval::Vector{Tv}
 end
 
+Base.size(A::MKLSparse.SparseMatrixCOO) = (A.m, A.n)
+Base.size(A::MKLSparse.SparseMatrixCSR) = (A.m, A.n)
+
 SparseArrays.nnz(A::MKLSparse.SparseMatrixCOO) = length(A.vals)
 SparseArrays.nnz(A::MKLSparse.SparseMatrixCSR) = length(A.nzval)