Add special casing for broadcasting a scalar to fix #104.

Author: chriselrod

src/LoopVectorization.jl

@@ -45,7 +45,6 @@ include("add_stores.jl")
 include("add_compute.jl")
 include("add_constants.jl")
 include("add_ifelse.jl")
-include("broadcast.jl")
 include("determinestrategy.jl")
 include("lower_compute.jl")
 include("lower_constant.jl")
@@ -59,6 +58,7 @@ include("reconstruct_loopset.jl")
 include("constructors.jl")
 include("user_api_conveniences.jl")
 include("mapreduce.jl")
+include("broadcast.jl")
 
 """
 LoopVectorization provides macros and functions that combine SIMD vectorization and

src/broadcast.jl

@@ -289,6 +289,25 @@ end
     q
     # ls
 end
+function vmaterialize!(
+    dest::StridedArray{T}, bc::Broadcasted{Base.Broadcast.DefaultArrayStyle{0},Nothing,typeof(identity),Tuple{T2}}, ::Val{Mod}
+) where {T <: NativeTypes, T2 <: Number, Mod}
+    arg = T(first(bc.args))
+    @avx for i ∈ eachindex(dest)
+        dest[i] = arg
+    end
+    dest
+end
+function vmaterialize!(
+    dest′::Union{Adjoint{T,A},Transpose{T,A}}, bc::Broadcasted{Base.Broadcast.DefaultArrayStyle{0},Nothing,typeof(identity),Tuple{T2}}, ::Val{Mod}
+) where {T <: NativeTypes, A <: StridedArray{T}, T2 <: Number, Mod}
+    arg = T(first(bc.args))
+    dest = parent(dest′)
+    @avx for i ∈ eachindex(dest)
+        dest[i] = arg
+    end
+    dest′
+end
 
 @inline function vmaterialize(bc::Broadcasted, ::Val{Mod}) where {Mod}
     ElType = Base.Broadcast.combine_eltypes(bc.f, bc.args)