Manual loop faster than broadcast

julia/sr.jl

@@ -285,7 +285,7 @@ function evalTreeArray(tree::Node, cX::Array{Float32, 2})::Union{Array{Float32,
             return nothing
         end
         op_idx = tree.op
-        UNAOP!(cumulator, op_idx)
+        UNAOP!(cumulator, op_idx, clen)
         @inbounds for i=1:clen
             if isinf(cumulator[i]) || isnan(cumulator[i])
                 return nothing
@@ -302,7 +302,7 @@ function evalTreeArray(tree::Node, cX::Array{Float32, 2})::Union{Array{Float32,
             return nothing
         end
         op_idx = tree.op
-        BINOP!(cumulator, array2, op_idx)
+        BINOP!(cumulator, array2, op_idx, clen)
         @inbounds for i=1:clen
             if isinf(cumulator[i]) || isnan(cumulator[i])
                 return nothing

pysr/sr.py

@@ -287,26 +287,34 @@ const limitPowComplexity = {"true" if limitPowComplexity else "false"}
     op_runner = ""
     if len(binary_operators) > 0:
         op_runner += """
-function BINOP!(x::Array{Float32, 1}, y::Array{Float32, 1}, i::Int)::Array{Float32, 1}
-    if i == 1
-        x .= """f"{binary_operators[0]}"".(x, y)"
+@inline function BINOP!(x::Array{Float32, 1}, y::Array{Float32, 1}, i::Int, clen::Int)
+    if i === 1
+        @inbounds @simd for j=1:clen
+            x[j] = """f"{binary_operators[0]}""""(x[j], y[j])
+        end"""
         for i in range(1, len(binary_operators)):
             op_runner += f"""
-    elseif i == {i+1}
-        x .= {binary_operators[i]}.(x, y)"""
+    elseif i === {i+1}
+        @inbounds @simd for j=1:clen
+            x[j] = {binary_operators[i]}(x[j], y[j])
+        end"""
         op_runner += """
     end
 end"""
 
     if len(unary_operators) > 0:
         op_runner += """
-function UNAOP!(x::Array{Float32, 1}, i::Int)::Array{Float32, 1}
-    if i == 1
-        x .= """f"{unary_operators[0]}.(x)"
+@inline function UNAOP!(x::Array{Float32, 1}, i::Int, clen::Int)
+    if i === 1
+        @inbounds @simd for j=1:clen
+            x[j] = """f"{unary_operators[0]}(x[j])""""
+        end"""
         for i in range(1, len(unary_operators)):
             op_runner += """
-    elseif i == {i+1}
-        x .= """f"{unary_operators[i]}.(x)"
+    elseif i === {i+1}
+        @inbounds @simd for j=1:clen
+            x[j] = {unary_operators[i]}(x[j])
+        end"""
         op_runner += """
     end
 end"""