enable Mooncake on 1.12 (#424)

* Enable Enzyme tests that are fixed now

* enable Mooncake

Author: penelopeysm

test/Project.toml

@@ -17,6 +17,7 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LogDensityProblems = "6fdf6af0-433a-55f7-b3ed-c6c6e0b8df7c"
 LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
 MCMCDiagnosticTools = "be115224-59cd-429b-ad48-344e309966f0"
+Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"

test/runtests.jl

@@ -9,6 +9,7 @@ using FiniteDifferences
 using ForwardDiff
 using Functors
 using LogExpFunctions
+using Mooncake
 using ReverseDiff
 using Pkg
 
@@ -31,23 +32,21 @@ using InverseFunctions: InverseFunctions
 using LazyArrays: LazyArrays
 
 const GROUP = get(ENV, "GROUP", "All")
-# Mooncake and Enzyme don't work on 1.12 yet
-const TEST_ENZYME_AND_MOONCAKE = VERSION < v"1.12.0"
+# Enzyme doesn't work on 1.12 yet
+const TEST_ENZYME = VERSION < v"1.12.0"
 
 TEST_ADTYPES = [
     ("ForwardDiff", AutoForwardDiff()),
     ("ReverseDiff", AutoReverseDiff(; compile=false)),
     ("ReverseDiffCompiled", AutoReverseDiff(; compile=true)),
+    ("Mooncake", AutoMooncake()),
 ]
-if TEST_ENZYME_AND_MOONCAKE
+if TEST_ENZYME
     Pkg.add("Enzyme")
     Pkg.add("EnzymeTestUtils")
-    Pkg.add("Mooncake")
     using Enzyme: Enzyme, set_runtime_activity, Forward, Reverse, Const
-    using Mooncake
     TEST_ADTYPES = [
         TEST_ADTYPES...,
-        ("Mooncake", AutoMooncake()),
         (
             "EnzymeForward",
             AutoEnzyme(; mode=set_runtime_activity(Forward), function_annotation=Const),
@@ -83,9 +82,9 @@ end
 if GROUP == "All" || GROUP == "AD"
     # These tests specifically check the implementation of AD backend rules.
     include("ad/chainrules.jl")
-    if TEST_ENZYME_AND_MOONCAKE
+    include("ad/mooncake.jl")
+    if TEST_ENZYME
         include("ad/enzyme.jl")
-        include("ad/mooncake.jl")
     end
 
     # These tests check that AD can differentiate through Bijectors