Merge pull request #248 from JuliaArrays/broadcastsize

Fix size of broadcast objects

Author: chriselrod

Project.toml

@@ -1,6 +1,6 @@
 name = "ArrayInterface"
 uuid = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-version = "5.0.0"
+version = "5.0.1"
 
 [deps]
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"

src/size.jl

@@ -17,6 +17,8 @@ julia> ArrayInterface.size(A)
 ```
 """
 size(a::A) where {A} = _maybe_size(Base.IteratorSize(A), a)
+size(a::Base.Broadcast.Broadcasted) = map(static_length, axes(a))
+
 _maybe_size(::Base.HasShape{N}, a::A) where {N,A} = map(static_length, axes(a))
 _maybe_size(::Base.HasLength, a::A) where {A} = (static_length(a),)
 size(x::SubArray) = eachop(_sub_size, to_parent_dims(x), x.indices)

test/runtests.jl

@@ -260,6 +260,7 @@ ArrayInterface.parent_type(::Type{DenseWrapper{T,N,P}}) where {T,N,P} = P
 
 @testset "Memory Layout" begin
     x = zeros(100);
+    @test ArrayInterface.size(Base.Broadcast.Broadcasted(+, (x, x'))) === (100,100)
     # R = reshape(view(x, 1:100), (10,10));
     # A = zeros(3,4,5);
     A = Wrapper(reshape(view(x, 1:60), (3,4,5)));