From 3f096f85c32b0305a128801093c1e10eb157c208 Mon Sep 17 00:00:00 2001 From: Penelope Yong Date: Wed, 6 Nov 2024 22:29:25 +0000 Subject: [PATCH] Disable more tests --- test/ad/flows.jl | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/test/ad/flows.jl b/test/ad/flows.jl index 100c264b..d2ebb49d 100644 --- a/test/ad/flows.jl +++ b/test/ad/flows.jl @@ -1,18 +1,20 @@ @testset "PlanarLayer" begin if v"1.11" <= VERSION < v"1.12" - broken = (:EnzymeForwardCrash,) + broken1 = (:EnzymeReverseCrash,) + broken2 = (:EnzymeForwardCrash, :EnzymeReverseCrash) else - broken = () + broken1 = () + broken2 = () end # logpdf of a flow with a planar layer and two-dimensional inputs - test_ad(randn(7)) do θ + test_ad(randn(7), broken1) do θ layer = PlanarLayer(θ[1:2], θ[3:4], θ[5:5]) flow = transformed(MvNormal(zeros(2), I), layer) x = θ[6:7] return logpdf(flow.dist, x) - logabsdetjac(flow.transform, x) end - test_ad(randn(11), broken) do θ + test_ad(randn(11), broken2) do θ layer = PlanarLayer(θ[1:2], θ[3:4], θ[5:5]) flow = transformed(MvNormal(zeros(2), I), layer) x = reshape(θ[6:end], 2, :) @@ -20,13 +22,13 @@ end # logpdf of a flow with the inverse of a planar layer and two-dimensional inputs - test_ad(randn(7)) do θ + test_ad(randn(7), broken1) do θ layer = PlanarLayer(θ[1:2], θ[3:4], θ[5:5]) flow = transformed(MvNormal(zeros(2), I), inverse(layer)) x = θ[6:7] return logpdf(flow.dist, x) - logabsdetjac(flow.transform, x) end - test_ad(randn(11), broken) do θ + test_ad(randn(11), broken2) do θ layer = PlanarLayer(θ[1:2], θ[3:4], θ[5:5]) flow = transformed(MvNormal(zeros(2), I), inverse(layer)) x = reshape(θ[6:end], 2, :)