cooperate with ArrayLayouts #153

JuliaArrays · Jul 18, 2023 · b1378a3 · b1378a3
1 parent 4eda5c7
commit b1378a3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,3 +4,4 @@
 deps/deps.jl
 .DS_Store
 Manifest.toml
+statprof/*
diff --git a/test/lazymultests.jl b/test/lazymultests.jl
@@ -17,6 +17,8 @@ Base.convert(::Type{MyMatrix{T}}, A::AbstractArray{T}) where T = MyMatrix{T}(A)
 Base.convert(::Type{MyMatrix{T}}, A::AbstractArray) where T = MyMatrix{T}(convert(AbstractArray{T}, A))
 Base.convert(::Type{MyMatrix}, A::AbstractArray{T}) where T = MyMatrix{T}(A)
 Base.getindex(A::MyMatrix, kj...) = A.A[kj...]
+Base.getindex(A::MyMatrix, k::Integer, j::AbstractVector) = A.A[k, j]
+Base.getindex(A::MyMatrix, k::AbstractVector, j::Integer) = A.A[k, j]
 Base.getindex(A::MyMatrix, k::Integer, ::Colon) = A.A[k,:]
 Base.getindex(A::MyMatrix, ::Colon, j::Integer) = A.A[:,j]
 Base.getindex(A::MyMatrix, ::Colon, j::AbstractVector) = MyMatrix(A.A[:,j])