Merge pull request #198 from JuliaArrays/staticfirstreinterpretaxes

have axes(reinterpret(reshape,T,::AbstractArray{S}),1) be static sometimes
JuliaArrays · Aug 31, 2021 · f8d1182 · f8d1182 · chriselrod · Aug 31, 2021
2 parents bb56a11 + 47096bf
commit f8d1182
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "ArrayInterface"
 uuid = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-version = "3.1.29"
+version = "3.1.30"
 
 [deps]
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"

diff --git a/src/axes.jl b/src/axes.jl
@@ -148,7 +148,12 @@ end
 
 
 @inline _axes(A::SubArray, dim::Integer) = Base.axes(A, Int(dim))  # TODO implement ArrayInterface version
-@inline _axes(A::ReinterpretArray, dim::Integer) = Base.axes(A, Int(dim))  # TODO implement ArrayInterface version
+@inline function _axes(A::ReinterpretArray{T,N,S}, dim::Integer) where {T,N,S}
+  if _is_reshaped(typeof(A)) && (sizeof(S) > sizeof(T)) && dim == 1
+    return One():static(div(sizeof(S), sizeof(T)))
+  end
+  Base.axes(A, Int(dim))  # TODO implement ArrayInterface version
+end
 @inline _axes(A::Base.ReshapedArray, dim::Integer) = Base.axes(A, Int(dim))  # TODO implement ArrayInterface version
 
 """