Redirect to base broadcast for DefaultArrayStyle{0}, fixes #319

Author: chriselrod

Project.toml

@@ -1,7 +1,7 @@
 name = "LoopVectorization"
 uuid = "bdcacae8-1622-11e9-2a5c-532679323890"
 authors = ["Chris Elrod <[email protected]>"]
-version = "0.12.64"
+version = "0.12.65"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

src/broadcast.jl

@@ -408,9 +408,9 @@ end
   storeop = add_simple_store!(ls, :destination, ArrayReference(:dest, loopsyms), elementbytes)
   doaddref!(ls, storeop)
   resize!(ls.loop_order, num_loops(ls)) # num_loops may be greater than N, eg Product
+  # return ls
   sc = setup_call(ls, :(Base.Broadcast.materialize!(dest, bc)), LineNumberNode(0), inline, false, u₁, u₂, v, threads%Int, warncheckarg)
   Expr(:block, Expr(:meta,:inline), sc, :dest)
-  # setup_call_debug(ls)
 end
 @generated function vmaterialize!(
     dest′::Union{Adjoint{T,A},Transpose{T,A}}, bc::BC, ::Val{Mod}, ::Val{UNROLL}
@@ -460,6 +460,9 @@ end
   end
 end
 
+@inline function vmaterialize(bc::Broadcasted{Base.Broadcast.DefaultArrayStyle{0}}, ::Val{Mod}, ::Val{UNROLL}) where {Mod,UNROLL}
+  Base.materialize(bc)
+end
 @inline function vmaterialize(
     bc::Broadcasted, ::Val{Mod}, ::Val{UNROLL}
 ) where {Mod,UNROLL}