update test to include mooncake

zuhengxu · zuhengxu · commit 91202ff6fc69 · 2025-03-04T10:42:03.000-08:00
diff --git a/test/Project.toml b/test/Project.toml
@@ -15,3 +15,6 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
+
+[compat]
+Mooncake = "0.4.101"
diff --git a/test/ad.jl b/test/ad.jl
@@ -29,7 +29,7 @@ end
         ADTypes.AutoForwardDiff(),
         ADTypes.AutoReverseDiff(; compile = false),
         ADTypes.AutoEnzyme(mode=Enzyme.set_runtime_activity(Enzyme.Reverse)),
-        # ADTypes.AutoMooncake(; config=Mooncake.Config()),
+        ADTypes.AutoMooncake(; config=Mooncake.Config()),
     ]
         @testset "$T" for T in [Float32, Float64]
             μ = 10 * ones(T, 2)
diff --git a/test/interface.jl b/test/interface.jl
@@ -6,7 +6,7 @@
         ADTypes.AutoForwardDiff(),
         ADTypes.AutoReverseDiff(),
         ADTypes.AutoEnzyme(mode=Enzyme.set_runtime_activity(Enzyme.Reverse)),
-        # ADTypes.AutoMooncake(; config = Mooncake.Config()), # somehow Mooncake does not work with Float64
+        ADTypes.AutoMooncake(; config = Mooncake.Config()), 
     ]
         @testset "$T" for T in [Float32, Float64]
             μ = 10 * ones(T, 2)
@@ -67,7 +67,7 @@ end
 #         ADTypes.AutoForwardDiff(),
 #         ADTypes.AutoReverseDiff(),
 #         ADTypes.AutoEnzyme(mode=Enzyme.set_runtime_activity(Enzyme.Reverse)),
-#         # ADTypes.AutoMooncake(; config = Mooncake.Config()), # somehow Mooncake does not work with Float64
+#         ADTypes.AutoMooncake(; config = Mooncake.Config()), # somehow Mooncake does not work with Float64
 #     ]
 #         @testset "$T" for T in [Float32, Float64]
 #             μ = 10 * ones(T, 2)

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ end`
`29`	`29`	`ADTypes.AutoForwardDiff(),`
`30`	`30`	`ADTypes.AutoReverseDiff(; compile = false),`
`31`	`31`	`ADTypes.AutoEnzyme(mode=Enzyme.set_runtime_activity(Enzyme.Reverse)),`
`32`		`- # ADTypes.AutoMooncake(; config=Mooncake.Config()),`
	`32`	`+ ADTypes.AutoMooncake(; config=Mooncake.Config()),`
`33`	`33`	`]`
`34`	`34`	`@testset "$T" for T in [Float32, Float64]`
`35`	`35`	`μ = 10 * ones(T, 2)`