Skip to content

Commit 3e1bd9c

Browse files
committed
Test rule
1 parent 9c838f3 commit 3e1bd9c

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

  • DifferentiationInterface/test/Back/DifferentiateWith

DifferentiationInterface/test/Back/DifferentiateWith/test.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,13 +62,17 @@ end;
6262
@testset for scen in filter(differentiatewith_scenarios()) do scen
6363
DIT.operator(scen) == :pullback
6464
end
65-
Mooncake.TestUtils.test_rule(StableRNG(0), scen.f, scen.x; is_primitive=true)
65+
Mooncake.TestUtils.test_rule(
66+
StableRNG(0), scen.f, scen.x; is_primitive=true, mode=Mooncake.ReverseMode()
67+
)
6668
end
6769
end;
6870

6971
@testset "Mooncake errors" begin
7072
MooncakeDifferentiateWithError =
71-
Base.get_extension(DifferentiationInterface, :DifferentiationInterfaceMooncakeExt).MooncakeDifferentiateWithError
73+
Base.get_extension(
74+
DifferentiationInterface, :DifferentiationInterfaceMooncakeExt
75+
).MooncakeDifferentiateWithError
7276

7377
e = MooncakeDifferentiateWithError(identity, 1.0, 2.0)
7478
@test sprint(showerror, e) ==

0 commit comments

Comments
 (0)