Have _vbroadcast on Bool return Mask

Author: chriselrod

Project.toml

@@ -1,7 +1,7 @@
 name = "VectorizationBase"
 uuid = "3d5dd08c-fd9d-11e8-17fa-ed2836048c2f"
 authors = ["Chris Elrod <[email protected]>"]
-version = "0.21.62"
+version = "0.21.63"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

src/llvm_intrin/masks.jl

@@ -1124,3 +1124,4 @@ end
 @inline Base.min(x::AbstractMask, y::AbstractMask) = x & y
 @inline Base.FastMath.max_fast(x::AbstractMask, y::AbstractMask) = x | y
 @inline Base.FastMath.min_fast(x::AbstractMask, y::AbstractMask) = x & y
+@inline zero_mask(::Type{T}) where {T} = zero_mask(zero(T))

src/llvm_intrin/vbroadcast.jl

@@ -110,6 +110,15 @@ end
   ::StaticInt{RS},
   ::True
 ) where {W,RS} = vbroadcast_expr(W, "half", :Float16, 2, RS)
+@inline function _vbroadcast(
+  ::StaticInt{W},
+  s::Bool,
+  ::StaticInt{RS}
+) where {W,RS}
+  t = Mask(max_mask(StaticInt{W}()))
+  f = Mask(zero_mask(StaticInt{W}()))
+  Core.ifelse(s, t, f)
+end
 @generated function _vbroadcast(
   ::StaticInt{W},
   s::_T,