Update test/ad/distributions.jl

TuringLang · Aug 21, 2024 · 443c731 · 443c731
1 parent 48c9afb
commit 443c731
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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),