diff --git a/test/ad/distributions.jl b/test/ad/distributions.jl index 66c910b..699bc7a 100644 --- a/test/ad/distributions.jl +++ b/test/ad/distributions.jl @@ -62,8 +62,8 @@ DistSpec(Poisson, (0.5,), [1, 1]), # Enzyme: no forward rule for ccall - DistSpec(Skellam, (1.0, 2.0), -2, broken=(:Zygote,)), - DistSpec(Skellam, (1.0, 2.0), [-2, -2], broken=(:Zygote,)), + DistSpec(Skellam, (1.0, 2.0), -2, broken=(:Enzyme,)), + DistSpec(Skellam, (1.0, 2.0), [-2, -2], broken=(:Enzyme,)), DistSpec(PoissonBinomial, ([0.5, 0.5],), 0),