Plop

Author: gdalle

Project.toml

@@ -7,7 +7,6 @@ version = "0.9.0"
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 DifferentiationInterface = "a0c0ee7d-e4b9-4e03-894e-1c5f64a51d63"
 Krylov = "ba0b0d4f-ebba-5204-a429-3ac8c609bfb7"
-LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LinearMaps = "7a12625a-238d-50fd-b39a-03d52299707e"
 LinearOperators = "5c8ed15e-5a4c-59e4-a42b-c7e8811fb125"
 
@@ -35,7 +34,6 @@ ForwardDiff = "0.10.36, 1"
 JET = "0.9, 0.10"
 JuliaFormatter = "2.1.2"
 Krylov = "0.9.6, 0.10"
-LinearAlgebra = "1.10"
 LinearMaps = "3.11.4"
 LinearOperators = "2.8.0"
 NLsolve = "4.5.1"

test/utils.jl

@@ -1,12 +1,12 @@
 using ADTypes
-using ADTypes: ForwardMode, ReverseMode
+using ADTypes: ForwardMode, ReverseMode, ForwardOrReverseMode
 using ChainRulesCore
 using ChainRulesTestUtils
 using ComponentArrays
 import DifferentiationInterface as DI
 using ForwardDiff: ForwardDiff
 import ImplicitDifferentiation as ID
-using ImplicitDifferentiation: ImplicitFunction
+using ImplicitDifferentiation: ImplicitFunction, prepare_implicit
 using JET
 using LinearAlgebra
 using Random: rand!
@@ -165,7 +165,7 @@ function test_implicit_rrule(scen::Scenario)
             @test dx ≈ dx_true
         end
         @testset "Unprepared" begin
-            (y, z), pb = rrule(ZygoteRuleConfig(), implicit, scen.x, scen.args...)
+            (y, z), pb = rrule_via_ad(ZygoteRuleConfig(), implicit, scen.x, scen.args...)
             dimpl, dx = pb((dy, dz))
             @test y ≈ y_true
             @test z == z_true