Restore linear indexing methods

JuliaStats · Dec 8, 2023 · 5710a1b · 5710a1b
1 parent c067537
commit 5710a1b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/pdmat.jl b/src/pdmat.jl
@@ -51,6 +51,10 @@ Base.broadcastable(a::PDMat) = Base.broadcastable(a.mat)
 
 ### Inheriting from AbstractMatrix
 
+Base.IndexStyle(::Type{PDMat{T,S}}) where {T,S} = Base.IndexStyle(S)
+# Linear Indexing
+Base.@propagate_inbounds Base.getindex(a::PDMat, i::Int) = getindex(a.mat, i)
+# Cartesian Indexing
 Base.@propagate_inbounds Base.getindex(a::PDMat, I::Vararg{Int, 2}) = getindex(a.mat, I...)
 
 ### Arithmetics

diff --git a/src/pdsparsemat.jl b/src/pdsparsemat.jl
@@ -51,7 +51,11 @@ LinearAlgebra.cholesky(a::PDSparseMat) = a.chol
 
 ### Inheriting from AbstractMatrix
 
-Base.getindex(a::PDSparseMat, I::Vararg{Int, 2}) = getindex(a.mat, I...)
+Base.IndexStyle(::Type{PDSparseMat{T,S}}) where {T,S} = IndexStyle(S)
+# Linear Indexing
+Base.@propagate_inbounds Base.getindex(a::PDSparseMat, i::Int) = getindex(a.mat, i)
+# Cartesian Indexing
+Base.@propagate_inbounds Base.getindex(a::PDSparseMat, I::Vararg{Int, 2}) = getindex(a.mat, I...)
 
 ### Arithmetics