Add outer reductions to check_args.

Author: chriselrod

Project.toml

@@ -20,7 +20,7 @@ IfElse = "0.1"
 OffsetArrays = "1"
 SLEEFPirates = "0.6"
 UnPack = "1"
-VectorizationBase = "0.13.7"
+VectorizationBase = "0.13.8"
 julia = "1.5"
 
 [extras]

src/condense_loopset.jl

@@ -264,6 +264,7 @@ end
 @inline check_args(_) = false
 @inline check_args(A, B, C::Vararg{Any,K}) where {K} = check_args(A) && check_args(B, C...)
 @inline check_args(::AbstractRange{T}) where {T} = check_type(T)
+@inline check_args(::Type{T}) where {T <: VectorizationBase.NativeTypesV} = true
 """
     check_type(::Type{T}) where {T}
 
@@ -275,6 +276,9 @@ check_type(::Type{T}) where {T} = false
 function check_args_call(ls::LoopSet)
     q = Expr(:call, lv(:check_args))
     append!(q.args, ls.includedactualarrays)
+    for r ∈ ls.outer_reductions
+        push!(q.args, Expr(:call, :typeof, name(ls.operations[r])))
+    end
     q
 end