Merge branch 'master' into MM/vecindex

Author: ararslan

perf/benchmarks.jl

@@ -0,0 +1,107 @@
+using BenchmarkTools, Base.Cartesian
+using Interpolations
+
+const suite = BenchmarkGroup()
+
+suite["bsplines"] = BenchmarkGroup()
+for bspline in ["constant", "linear", "quadratic", "cubic"]
+    suite["bsplines"][bspline] = BenchmarkGroup()
+end
+
+# To evaluate at fractional positions without any "unnecessary"
+# overhead, safer to use Base.Cartesian
+@generated function sumvalues{T,N}(itp::AbstractInterpolation{T,N}, inds)
+    quote
+        @nexprs $N d->inds_d = inds[d]
+        s = zero(eltype(itp))
+        @inbounds @nloops $N i d->inds_d begin
+            s += @nref($N, itp, i)
+        end
+        s
+    end
+end
+
+function sumvalues_indices(itp)
+    inds = indices(itp)
+    n = Int(round(10^(3/ndims(itp))))
+    ntuple(d->collect(linspace(first(inds[d])+0.001, last(inds[d])-0.001, n)), ndims(itp))
+end
+
+strip_prefix(str) = replace(str, "Interpolations.", "")
+benchstr{T<:Interpolations.GridType}(::Type{T}) = strip_prefix(string(T))
+
+benchstr(::Type{Constant}) = "Constant()"
+benchstr(::Type{Linear}) = "Linear()"
+benchstr{BC<:Interpolations.Flag}(::Type{Quadratic{BC}}) =
+    string("Quadratic(", strip_prefix(string(BC)), "())")
+benchstr{BC<:Interpolations.Flag}(::Type{Cubic{BC}}) =
+    string("Quadratic(", strip_prefix(string(BC)), "())")
+
+groupstr(::Type{Constant}) = "constant"
+groupstr(::Type{Linear}) = "linear"
+groupstr(::Type{Quadratic}) = "quadratic"
+groupstr(::Type{Cubic}) = "cubic"
+
+
+for A in (collect(Float64, 1:3),
+          reshape(collect(Float64, 1:9), 3, 3),
+          reshape(collect(Float64, 1:27), 3, 3, 3))
+    # Constant & Linear
+    for D in (Constant, Linear)
+        gstr = groupstr(D)
+        for GT in (OnGrid, OnCell)
+            Ac = copy(A)
+            idstr = string(ndims(A), "d_", benchstr(D), '_', benchstr(GT))
+            suite["bsplines"][gstr][string(idstr, "_construct")] =
+                @benchmarkable interpolate($Ac, BSpline($D()), $GT())
+            itp = interpolate(copy(A), BSpline(D()), GT())
+            inds = sumvalues_indices(itp)
+            suite["bsplines"][gstr][string(idstr, "_use")] =
+                @benchmarkable sumvalues($itp, $inds)
+        end
+    end
+    # Quadratic
+    gstr = groupstr(Quadratic)
+    for BC in (Flat,Line,Free,Periodic,Reflect,Natural), GT in (OnGrid, OnCell)
+        Ac = copy(A)
+        idstr = string(ndims(A), "d_", benchstr(Quadratic{BC}), '_', benchstr(GT))
+        suite["bsplines"][gstr][string(idstr, "_construct")] =
+            @benchmarkable interpolate($Ac, BSpline(Quadratic($BC())), $GT())
+        itp = interpolate(copy(A), BSpline(Quadratic(BC())), GT())
+        inds = sumvalues_indices(itp)
+        suite["bsplines"][gstr][string(idstr, "_use")] =
+            @benchmarkable sumvalues($itp, $inds)
+    end
+    for BC in (InPlace,InPlaceQ)
+        Ac = copy(A)
+        idstr = string(ndims(A), "d_", benchstr(Quadratic{BC}), '_', benchstr(OnCell))
+        suite["bsplines"][gstr][string(idstr, "_construct")] =
+            @benchmarkable interpolate!($Ac, BSpline(Quadratic($BC())), OnCell())
+        itp = interpolate!(copy(A), BSpline(Quadratic(BC())), OnCell())
+        inds = sumvalues_indices(itp)
+        suite["bsplines"][gstr][string(idstr, "_use")] =
+            @benchmarkable sumvalues($itp, $inds)
+    end
+    # Cubic
+    gstr = groupstr(Cubic)
+    for BC in (Flat,Line,Free,Periodic), GT in (OnGrid, OnCell)
+        Ac = copy(A)
+        idstr = string(ndims(A), "d_", benchstr(Cubic{BC}), '_', benchstr(GT))
+        suite["bsplines"][gstr][string(idstr, "_construct")] =
+            @benchmarkable interpolate($Ac, BSpline(Cubic($BC())), $GT())
+        itp = interpolate(copy(A), BSpline(Cubic(BC())), GT())
+        inds = sumvalues_indices(itp)
+        suite["bsplines"][gstr][string(idstr, "_use")] =
+            @benchmarkable sumvalues($itp, $inds)
+    end
+end
+
+paramspath = joinpath(dirname(@__FILE__), "params.jld")
+
+if isfile(paramspath)
+    loadparams!(suite, BenchmarkTools.load(paramspath, "suite"), :evals);
+else
+    info("Tuning suite (this may take a while)")
+    tune!(suite)
+    BenchmarkTools.save(paramspath, "suite", params(suite));
+end