WIP: Faster Inference (#357)

We need to check if this regresses quality somewhere, but if not, this
should speed up inference of LoopVectorization a fair bit.

Author: Keno

src/codegen/lower_memory_common.jl

@@ -13,7 +13,7 @@ function symbolind(ind::Symbol, op::Operation, td::UnrollArgs, ls::LoopSet)
     Symbol(pvar, '_', Core.ifelse(u₁op, u₁, 1)), parent
 end
 
-staticexpr(x::Int) = Expr(:call, Expr(:curly, lv(:StaticInt), x))
+staticexpr(x::Int) = StaticInt{x}()
 staticexpr(x::Union{Symbol,Expr}) = Expr(:call, lv(:StaticInt), x)
 
 _MMind(ind::Union{Symbol,Expr}, str::Int) = Expr(:call, lv(:MM), VECTORWIDTHSYMBOL, ind, staticexpr(str))

src/codegen/lowering.jl

@@ -474,7 +474,7 @@ end
     S
   end
 end
-function outer_reduction_zero(op::Operation, u₁u::Bool, Umax::Int, reduct_class::Float64, rs::Expr)
+function outer_reduction_zero(op::Operation, u₁u::Bool, Umax::Int, reduct_class::Float64, rs::Union{Expr, StaticInt})
   isifelse = instruction(op).instr === :ifelse
   reduct_zero = if isifelse
     Symbol(name(op), "##BASE##EXTRACT##")
@@ -515,7 +515,7 @@ end
 
 # TODO: handle tiled outer reductions; they will require a suffix arg
 function initialize_outer_reductions!(
-  q::Expr, ls::LoopSet, op::Operation, _Umax::Int, us::UnrollSpecification, rs::Expr
+  q::Expr, ls::LoopSet, op::Operation, _Umax::Int, us::UnrollSpecification, rs::Union{Expr, StaticInt}
   )
   @unpack u₁, u₂ = us
   Umax = u₂ == -1 ? _Umax : u₁
@@ -541,7 +541,7 @@ function initialize_outer_reductions!(
   end
   nothing
 end
-function initialize_outer_reductions!(q::Expr, ls::LoopSet, Umax::Int)
+function initialize_outer_reductions!(q::Expr, ls::LoopSet, Umax::Union{Int, StaticInt})
     rs = staticexpr(reg_size(ls))
     us = ls.unrollspecification
     for or ∈ ls.outer_reductions