fix #424

Author: chriselrod

Project.toml

@@ -1,7 +1,7 @@
 name = "LoopVectorization"
 uuid = "bdcacae8-1622-11e9-2a5c-532679323890"
 authors = ["Chris Elrod <elrodc@gmail.com>"]
-version = "0.12.123"
+version = "0.12.124"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

src/modeling/graphs.jl

@@ -1646,12 +1646,17 @@ function Base.push!(
     @assert length(localbody.args) == 2
     LHS = (localbody.args[1])::Symbol
     RHS_1 = localbody.args[2]
-    RHS_1 isa Symbol && return push!(ls, localbody, elementbytes, position, mpref)
-    RHS = push!(ls, RHS_1, elementbytes, position, mpref)
-    if isstore(RHS)
-      RHS
+    if RHS_1 isa Symbol
+      return push!(ls, localbody, elementbytes, position, mpref)
+    elseif Meta.isexpr(RHS_1, :(=), 2)
+      RHS = push!(ls, RHS_1, elementbytes, position, mpref)
+      if isstore(RHS)
+        RHS
+      else
+        add_compute!(ls, LHS, :identity, [RHS], elementbytes)
+      end
     else
-      add_compute!(ls, LHS, :identity, [RHS], elementbytes)
+      push!(ls, localbody, elementbytes, position, mpref)
     end
   else
     throw(LoopError("Don't know how to handle expression.", ex))

test/ifelsemasks.jl

@@ -506,6 +506,15 @@ T = Float32
     end
     minval, indmin
   end
+  function extrema_turbo(x)
+    a = b = first(x)
+    @turbo for i in eachindex(x)
+      local e = x[i]
+      b = max(b, e)
+      a = min(a, e)
+    end
+    a, b
+  end
 
   N = 117
   for T ∈ (Float32, Float64, Int32, Int64)
@@ -536,6 +545,7 @@ T = Float32
     c1 = similar(a)
     c2 = similar(a)
 
+    @test extrema(a) == extrema_turbo(a)
     promote_bool_store!(c1, a, b)
     promote_bool_storeavx!(c2, a, b)
     @test c1 == c2