Allow conditions to be symbols instead of just expressions.

Author: chriselrod

src/add_ifelse.jl

@@ -8,7 +8,9 @@ function add_if!(ls::LoopSet, LHS::Symbol, RHS::Expr, elementbytes::Int, positio
     # for now, just simple 1-liners
     @assert length(RHS.args) == 3 "if statements without an else cannot be assigned to a variable."
     condition = first(RHS.args)
-    condop = if isnothing(mpref)
+    condop = if condition isa Symbol
+        getop(ls, condition, elementbytes)
+    elseif isnothing(mpref)
         add_operation!(ls, gensym(:mask), condition, elementbytes, position)
     else
         add_operation!(ls, gensym(:mask), condition, mpref, elementbytes, position)

test/ifelsemasks.jl

@@ -99,7 +99,8 @@ T = Float32
     end
     function addormulp1avx!(c, a, b)
         @avx for i ∈ eachindex(c,a,b)
-            c[i] = 1 + (a[i] > b[i] ? a[i] + b[i] : a[i] * b[i])
+            a_greater_than_b = a[i] > b[i]
+            c[i] = 1 + (a_greater_than_b ? a[i] + b[i] : a[i] * b[i])
         end
     end
     function addifelsemul_avx!(c, a, b)