Fix ArrayInterface.size for dynamically sized vectors, add test for that, and also fix known_length for reinterpreted static ranges.

Author: chriselrod

Project.toml

@@ -1,6 +1,6 @@
 name = "ArrayInterface"
 uuid = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-version = "2.14.8"
+version = "2.14.9"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

src/stridelayout.jl

@@ -422,6 +422,12 @@ end
 end
 
 
+@inline function known_length(::Type{T}) where {T <: Base.ReinterpretArray}
+    _known_length(known_length(parent_type(T)), eltype(T), eltype(parent_type(T)))
+end
+_known_length(::Nothing, _, __) = nothing
+@inline _known_length(L::Integer, ::Type{T}, ::Type{P}) where {T,P} = L * sizeof(P) ÷ sizeof(T)
+
 # These methods help handle identifying axes that dont' directly propagate from the
 # parent array axes. They may be worth making a formal part of the API, as they provide
 # a low traffic spot to change what axes_types produces.
@@ -617,7 +623,7 @@ end
     Expr(:block, Expr(:meta, :inline), t)
 end
 
-@inline size(v::AbstractVector) = (static_length(axes_types(v, 1)),)
+@inline size(v::AbstractVector) = (static_length(v),)
 @inline size(B::Union{Transpose{T,A},Adjoint{T,A}}) where {T,A<:AbstractMatrix{T}} =
     permute(size(parent(B)), Val{(2, 1)}())
 @inline size(B::PermutedDimsArray{T,N,I1,I2,A}) where {T,N,I1,I2,A<:AbstractArray{T,N}} =

test/runtests.jl

@@ -371,14 +371,16 @@ end
     Rr = reinterpret(Int32, R)
     Ar = reinterpret(Float32, A)
 
- 
+    sv5 = @SVector(zeros(5)); v5 = Vector{Float64}(undef, 5);
+    @test @inferred(ArrayInterface.size(sv5)) === (StaticInt(5),)
+    @test @inferred(ArrayInterface.size(v5)) === (5,)
     @test @inferred(ArrayInterface.size(A)) === (3,4,5)
     @test @inferred(ArrayInterface.size(Ap)) === (2,5)
     @test @inferred(ArrayInterface.size(A)) === size(A)
     @test @inferred(ArrayInterface.size(Ap)) === size(Ap)
     @test @inferred(ArrayInterface.size(R)) === (StaticInt(2),)
     @test @inferred(ArrayInterface.size(Rr)) === (StaticInt(4),)
-
+    @test @inferred(ArrayInterface.known_length(Rr)) === 4
 
     @test @inferred(ArrayInterface.size(S)) === (StaticInt(2), StaticInt(3), StaticInt(4))
     @test @inferred(ArrayInterface.size(Sp)) === (2, 2, StaticInt(3))