add benchmark group for solve_ROF_PD

JuliaImages · Oct 21, 2021 · 3c441fa · 3c441fa
1 parent a9ed869
commit 3c441fa
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/benchmark/benchmarks.jl b/benchmark/benchmarks.jl
@@ -2,6 +2,7 @@ using ImageFiltering, ImageCore
 using PkgBenchmark
 using BenchmarkTools
 using Statistics: quantile, mean, median!
+using ImageFiltering.Models
 
 function makeimages(sz)
     imgF32      = rand(Float32, sz)
@@ -56,3 +57,14 @@ let grp = SUITE["imfilter"]
         end
     end
 end
+
+
+SUITE["ROF"] = BenchmarkGroup()
+let grp = SUITE["ROF"]
+    for sz in ((100, 100), (2048, 2048), (2048,), (100, 100, 100))
+        for (aname, img) in makeimages(sz)
+            szstr = sz2str(sz)
+            grp["PrimalDual"*"_"*aname*"_"*szstr] = @benchmarkable solve_ROF_PD($img, 0.1, 10)
+        end
+    end
+end