fix benchmark disable Zygote

TuringLang · Jun 5, 2024 · c0a0a3b · c0a0a3b
1 parent 5d4c00c
commit c0a0a3b
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/bench/benchmarks.jl b/bench/benchmarks.jl
@@ -22,15 +22,16 @@ include("normallognormal.jl")
 
 const SUITES = BenchmarkGroup()
 
-SUITES["normal + bijector"]["meanfield"]["Zygote"] =
-    @benchmarkable normallognormal(
-        ;
-        fptype       = Float64,
-        adtype       = AutoZygote(),
-        family       = :meanfield,
-        objective    = :RepGradELBO,
-        n_montecarlo = 4,
-    )
+# Comment until https://github.com/TuringLang/Bijectors.jl/pull/315 is merged
+# SUITES["normal + bijector"]["meanfield"]["Zygote"] =
+#     @benchmarkable normallognormal(
+#         ;
+#         fptype       = Float64,
+#         adtype       = AutoZygote(),
+#         family       = :meanfield,
+#         objective    = :RepGradELBO,
+#         n_montecarlo = 4,
+#     )
 
 SUITES["normal + bijector"]["meanfield"]["ReverseDiff"] =
     @benchmarkable normallognormal(