renamed Zygote.forward to Zygote.pullback since v0.4

Author: torfjelde

src/common.jl

@@ -41,7 +41,7 @@ function turing_chol(A::AbstractMatrix, check)
 end
 turing_chol(A::Tracker.TrackedMatrix, check) = Tracker.track(turing_chol, A, check)
 Tracker.@grad function turing_chol(A::AbstractMatrix, check)
-    C, back = Zygote.forward(unsafe_cholesky, Tracker.data(A), Tracker.data(check))
+    C, back = Zygote.pullback(unsafe_cholesky, Tracker.data(A), Tracker.data(check))
     return (C.factors, C.info), Δ->back((factors=Tracker.data(Δ[1]),))
 end
 
@@ -104,7 +104,7 @@ function zygote_ldiv(A::Tracker.TrackedMatrix, B::AbstractVecOrMat)
 end
 zygote_ldiv(A::AbstractMatrix, B::TrackedVecOrMat) =  Tracker.track(zygote_ldiv, A, B)
 Tracker.@grad function zygote_ldiv(A, B)
-    Y, back = Zygote.forward(\, Tracker.data(A), Tracker.data(B))
+    Y, back = Zygote.pullback(\, Tracker.data(A), Tracker.data(B))
     return Y, Δ->back(Tracker.data(Δ))
 end
 

test/test_utils.jl

@@ -96,7 +96,7 @@ end
 
 Check that the reverse-mode sensitivities produced by an AD library are correct for `f`
 at `x...`, given sensitivity `ȳ` w.r.t. `y = f(x...)` up to `rtol` and `atol`.
-`forward` should be either `Tracker.forward` or `Zygote.forward`.
+`forward` should be either `Tracker.forward` or `Zygote.pullback`.
 """
 function test_reverse_mode_ad(forward, f, ȳ, x...; rtol=1e-8, atol=1e-8)