Fixed broadcast with unknown functions.

Author: chriselrod

benchmarks/driver.jl

@@ -2,6 +2,12 @@
 # const LOOPVECBENCHDIR = joinpath(pkgdir("LoopVectorization"), "benchmarks")
 # includet(joinpath(LOOPVECBENCHDIR, "driver.jl"))
 
+pkgdir(pkg::String) = abspath(joinpath(dirname(Base.find_package(pkg)), ".."))
+const LOOPVECBENCHDIR = joinpath(pkgdir("LoopVectorization"), "benchmarks")
+include(joinpath(LOOPVECBENCHDIR, "benchmarkflops.jl"))
+include(joinpath(LOOPVECBENCHDIR, "plotbenchmarks.jl"))
+
+
 using Distributed
 
 addprocs(9);
@@ -33,5 +39,23 @@ exp_bench = fetch(exp_future)
 aplusBc_bench = fetch(aplusBc_future)
 
 
-include(joinpath(LOOPVECBENCHDIR, "plotbenchmarks.jl"))
+plot(gemm_bench)
+plot(AtmulB_bench)
+plot(dot_bench)
+plot(selfdot_bench)
+plot(gemv_bench)
+plot(dot3_bench)
+plot(sse_bench)
+plot(exp_bench)
+plot(aplusBc_bench)
+
+save(joinpath("~/Pictures", "bench_gemm_v3.png"), plot(gemm_bench));
+save(joinpath("~/Pictures", "bench_AtmulB_v3.png"), plot(AtmulB_bench));
+save(joinpath("~/Pictures", "bench_dot_v3.png"), plot(dot_bench));
+save(joinpath("~/Pictures", "bench_selfdot_v3.png"), plot(selfdot_bench));
+save(joinpath("~/Pictures", "bench_gemv_v3.png"), plot(gemv_bench));
+save(joinpath("~/Pictures", "bench_dot3_v3.png"), plot(dot3_bench));
+save(joinpath("~/Pictures", "bench_sse_v3.png"), plot(sse_bench));
+save(joinpath("~/Pictures", "bench_exp_v3.png"), plot(exp_bench));
+save(joinpath("~/Pictures", "bench_aplusBc_v3.png"), plot(aplusBc_bench));
 

src/broadcast.jl

@@ -154,9 +154,7 @@ function add_broadcast!(
     elementbytes::Int = 8
 ) where {N,S<:Base.Broadcast.AbstractArrayStyle{N},F,A}
     instr = get(FUNCTIONSYMBOLS, F) do
-        f = gensym(:f)
-        pushpreamble!(ls, Expr(:(=), f, Expr(:(.), bcname, QuoteNode(:f))))
-        f
+        Instruction(bcname, :f)
     end
     args = A.parameters
     Nargs = length(args)

test/runtests.jl

@@ -529,6 +529,11 @@ end
         three = 3; fill!(b1, -9999.999);
         @avx @. b1 = three*a + sin(a) + sqrt(a);
         @test b1 ≈ b2
+
+        C = rand(100,10,10);
+        D1 = C .^ 0.3;
+        D2 = @avx C .^ 0.3;
+        @test D1 ≈ D2
     end
 end