Update distributions.jl

TuringLang · Aug 21, 2024 · 48c9afb · 48c9afb
1 parent 5d3f0c5
commit 48c9afb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/ad/distributions.jl b/test/ad/distributions.jl
@@ -61,8 +61,9 @@
         DistSpec(Poisson, (0.5,), 1),
         DistSpec(Poisson, (0.5,), [1, 1]),
 
-        DistSpec(Skellam, (1.0, 2.0), -2),
-        DistSpec(Skellam, (1.0, 2.0), [-2, -2]),
+        # 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(PoissonBinomial, ([0.5, 0.5],), 0),