treat anon funcs as opaque functions (instead of throwing)

Author: chriselrod

Project.toml

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

src/modeling/graphs.jl

@@ -971,21 +971,20 @@ end
 # end
 # instruction(ls::LoopSet, f::Symbol) = instruction!(ls, f)
 function instruction!(ls::LoopSet, x::Expr)
-    # x isa Symbol && return x
-    if x.head === :$
-        _x = only(x.args)
-        _x isa Symbol && return instruction!(ls, _x)
-        @assert _x isa Expr
-        x = _x
-    end
+  # x isa Symbol && return x
+  if x.head === :$
+    _x = only(x.args)
+    _x isa Symbol && return instruction!(ls, _x)
+    @assert _x isa Expr
+    x = _x
+  end
+  if x.head ≢ :(->)
     instr = last(x.args).value
-    if instr ∉ keys(COST)
-        instr = gensym!(ls, "f")
-        pushpreamble!(ls, Expr(:(=), instr, x))
-        Instruction(Symbol(""), instr)
-    else
-        Instruction(:LoopVectorization, instr)
-    end
+    instr ∈ keys(COST) && return Instruction(:LoopVectorization, instr)
+  end
+  instr = gensym!(ls, "f")
+  pushpreamble!(ls, Expr(:(=), instr, x))
+  Instruction(Symbol(""), instr)
 end
 instruction!(ls::LoopSet, x::Symbol) = instruction(x)
 function instruction!(ls::LoopSet, f::F) where {F <: Function}

test/miscellaneous.jl

@@ -735,7 +735,7 @@ function findreducedparentfornonvecstoreavx!(U::AbstractMatrix{T}, E1::AbstractV
             U[i,j] = t
             _s += a * (1 - t^2)
         end
-        E1[j] = _s / n
+        E1[j] = (x -> x / n)(_s)
     end
     U,E1
 end