cse getproperty in add_operation!. Fixes #217

Author: chriselrod

Project.toml

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

src/modeling/graphs.jl

@@ -970,6 +970,13 @@ function add_operation!(
         add_if!(ls, LHS, RHS, elementbytes, position)
     elseif RHS.head === :block
         add_operation!(ls, LHS, strip_op_linenumber_nodes(RHS), elementbytes, position)
+    elseif RHS.head === :(.)
+        c = gensym!(ls, "getproperty")
+        pushpreamble!(ls, Expr(:(=), c, RHS))
+        add_constant!(ls, c, elementbytes)
+        # op = add_constant!(ls, c, ls.loopsymbols[1:position], LHS, elementbytes, :numericconstant)
+        # pushpreamble!(ls, op, c)
+        # op
     else
         throw(LoopError("Expression not recognized.", RHS))
     end
@@ -1005,6 +1012,13 @@ function add_operation!(
         add_if!(ls, LHS_sym, RHS, elementbytes, position, LHS_ref)
     elseif RHS.head === :block
         add_operation!(ls, LHS, strip_op_linenumber_nodes(RHS), elementbytes, position)
+    elseif RHS.head === :(.)
+        c = gensym!(ls, "getproperty")
+        pushpreamble!(ls, Expr(:(=), c, RHS))
+        add_constant!(ls, c, elementbytes)
+        # op = add_constant!(ls, c, ls.loopsymbols[1:position], LHS_sym, elementbytes, :numericconstant)
+        # pushpreamble!(ls, op, c)
+        # op
     else
         throw(LoopError("Expression not recognized.", RHS))
     end

src/parse/add_constants.jl

@@ -56,10 +56,10 @@ function add_constant!(
     ls::LoopSet, value::Symbol, deps::Vector{Symbol}, assignedsym::Symbol, elementbytes::Int, f::Symbol = Symbol("")
 )
     retop = get(ls.opdict, value, nothing)
-    if retop !== nothing
-        op = Operation(length(operations(ls)), assignedsym, elementbytes, :identity, compute, deps, reduceddependencies(retop), [retop])
-    else
+    if retop === nothing
         op = Operation(length(operations(ls)), assignedsym, elementbytes, Instruction(f, value), constant, deps, NODEPENDENCY, NOPARENTS)
+    else
+        op = Operation(length(operations(ls)), assignedsym, elementbytes, :identity, compute, deps, reduceddependencies(retop), [retop])
     end
     pushop!(ls, op, assignedsym)
 end