Benchmark shared global Hessian-tracing

adrhill · Jul 26, 2024 · 10eb7a5 · 10eb7a5
1 parent ec185ef
commit 10eb7a5
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/benchmark/bench_jogger.jl b/benchmark/bench_jogger.jl
@@ -20,9 +20,10 @@ suite["OptimizationProblems"] = optbench([:britgas])
 for S1 in SET_TYPES
     S2 = Set{Tuple{Int,Int}}
 
+    # Non-shared tracers 
+    shared = false
     PG = IndexSetGradientPattern{Int,S1}
-    PH = IndexSetHessianPattern{Int,S1,S2,false}
-
+    PH = IndexSetHessianPattern{Int,S1,S2,shared}
     G = GradientTracer{PG}
     H = HessianTracer{PH}
 
@@ -34,4 +35,15 @@ for S1 in SET_TYPES
     suite["Hessian"]["Local"][(nameof(S1), nameof(S2))] = hessbench(
         TracerLocalSparsityDetector(G, H)
     )
+
+    # Shared tracers 
+    shared = true
+    PG = IndexSetGradientPattern{Int,S1}
+    PH = IndexSetHessianPattern{Int,S1,S2,shared}
+    G = GradientTracer{PG}
+    H = HessianTracer{PH}
+
+    suite["Hessian"]["Global (shared)"][(nameof(S1), nameof(S2))] = hessbench(
+        TracerSparsityDetector(G, H)
+    )
 end