Merge pull request #80 from jipolanco/fix-extradims

Fix getindex with extra dims
jipolanco · May 12, 2023 · 4f4b5f1 · 4f4b5f1
2 parents 15fa3f9 + f6fbe52
commit 4f4b5f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "PencilArrays"
 uuid = "0e08944d-e94e-41b1-9406-dcf66b6a9d2e"
 authors = ["Juan Ignacio Polanco <[email protected]> and contributors"]
-version = "0.17.11"
+version = "0.17.12"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

diff --git a/src/arrays.jl b/src/arrays.jl
@@ -330,14 +330,7 @@ end
     parent(x)[_genperm(x, I)...] = v
 
 @inline function _genperm(x::PencilArray{T,N}, I::NTuple{N,Int}) where {T,N}
-    # Split "spatial" and "extra" indices.
-    M = ndims_space(x)
-    E = ndims_extra(x)
-    @assert M + E === N
-    J = ntuple(n -> @inbounds(I[n]), Val(M))
-    K = ntuple(n -> @inbounds(I[M + n]), Val(E))
-    perm = permutation(x)
-    ((perm * J)..., K...)
+    permutation(x) * I
 end
 
 @inline _genperm(x::PencilArray, I::CartesianIndex) =