Merge pull request #1234 from SciML/ChrisRackauckas-patch-3

Update Mooncake error throw derivatives

Author: ChrisRackauckas

test/alternative_ad_frontend.jl

@@ -61,10 +61,8 @@ dup = Zygote.gradient(senseloss(InterpolatingAdjoint()), u0p)[1]
 @test_broken only(Enzyme.gradient(Reverse, senseloss(ForwardSensitivity()), u0p)) ≈ dup # broken because ForwardSensitivity not compatible with perturbing u0
 
 @test mooncake_gradient(senseloss(InterpolatingAdjoint()), u0p) ≈ dup
-@test_throws Any mooncake_gradient(senseloss(ReverseDiffAdjoint()), u0p) ≈ dup
-@test_throws Any mooncake_gradient(senseloss(TrackerAdjoint()), u0p) ≈ dup
-#@test_broken @test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss(ReverseDiffAdjoint()), u0p) ≈ dup
-#@test_broken @test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss(TrackerAdjoint()), u0p) ≈ dup
+@test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss(ReverseDiffAdjoint()), u0p)
+@test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss(TrackerAdjoint()), u0p)
 @test mooncake_gradient(senseloss(ForwardDiffSensitivity()), u0p) ≈ dup
 @test_broken mooncake_gradient(senseloss(ForwardSensitivity()), u0p) ≈ dup # broken because ForwardSensitivity not compatible with perturbing u0
 
@@ -103,10 +101,8 @@ dup = Zygote.gradient(senseloss2(InterpolatingAdjoint()), u0p)[1]
 @test_broken only(Enzyme.gradient(Reverse, senseloss2(ForwardSensitivity()), u0p)) ≈ dup # broken because ForwardSensitivity not compatible with perturbing u0
 
 @test mooncake_gradient(senseloss2(InterpolatingAdjoint()), u0p) ≈ dup
-@test_throws Any mooncake_gradient(senseloss2(ReverseDiffAdjoint()), u0p) ≈ dup
-@test_throws Any mooncake_gradient(senseloss2(TrackerAdjoint()), u0p) ≈ dup
-#@test_broken @test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss2(ReverseDiffAdjoint()), u0p) ≈ dup
-#@test_broken @test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss2(TrackerAdjoint()), u0p) ≈ dup
+@test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss2(ReverseDiffAdjoint()), u0p)
+@test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss2(TrackerAdjoint()), u0p)
 @test mooncake_gradient(senseloss2(ForwardDiffSensitivity()), u0p) ≈ dup
 @test_broken mooncake_gradient(senseloss2(ForwardSensitivity()), u0p) ≈ dup # broken because ForwardSensitivity not compatible with perturbing u0
 
@@ -143,10 +139,8 @@ dup = Zygote.gradient(senseloss3(InterpolatingAdjoint()), u0p)[1]
 @test_broken only(Enzyme.gradient(Reverse, senseloss3(ForwardSensitivity()), u0p)) ≈ dup
 
 @test mooncake_gradient(senseloss3(InterpolatingAdjoint()), u0p) ≈ dup
-@test_throws Any mooncake_gradient(senseloss3(ReverseDiffAdjoint()), u0p) ≈ dup
-@test_throws Any mooncake_gradient(senseloss3(TrackerAdjoint()), u0p) ≈ dup
-#@test_broken @test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss3(ReverseDiffAdjoint()), u0p) ≈ dup
-#@test_broken @test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss3(TrackerAdjoint()), u0p) ≈ dup
+@test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss3(ReverseDiffAdjoint()), u0p)
+@test_throws SciMLSensitivity.MooncakeTrackedRealError mooncake_gradient(senseloss3(TrackerAdjoint()), u0p)
 @test mooncake_gradient(senseloss3(ForwardDiffSensitivity()), u0p) ≈ dup
 @test_broken mooncake_gradient(senseloss3(ForwardSensitivity()), u0p) ≈ dup
 
@@ -289,4 +283,4 @@ grad_rd = ReverseDiff.gradient(loss2, p)
 @test grad_fd≈grad_fi atol=1e-2
 @test grad_fd≈grad_zg atol=1e-4
 @test grad_fd≈grad_rd atol=1e-4
-@test_broken mooncake_gradient(loss2, p) ≈ grad_rd atol=1e-4
+@test_broken mooncake_gradient(loss2, p) ≈ grad_rd atol=1e-4