faster sparse(::Diagonal)

JuliaLang · Aug 31, 2017 · e3b4bd0 · e3b4bd0
1 parent 7e35ac6
commit e3b4bd0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/base/sparse/sparsematrix.jl b/base/sparse/sparsematrix.jl
@@ -764,6 +764,11 @@ function sparse(B::Bidiagonal)
     return sparse([1:m;1:m-1],[1:m;2:m],[B.dv;B.ev], Int(m), Int(m)) # upper bidiagonal
 end
 
+function sparse(D::Diagonal)
+    m = length(D.diag)
+    return SparseMatrixCSC(m, m, [1:m; m+1], [1:m;], D.diag)
+end
+
 ## Transposition and permutation methods
 
 """