From c0a0a3bebc8fb8d44f0f683e889abcf796780c30 Mon Sep 17 00:00:00 2001 From: Ray Kim Date: Wed, 5 Jun 2024 23:50:00 +0100 Subject: [PATCH] fix benchmark disable Zygote --- bench/benchmarks.jl | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/bench/benchmarks.jl b/bench/benchmarks.jl index 03a95cff..8585fa8c 100644 --- 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(