Skip to content

Commit 462be07

Browse files
committed
Add Mooncake to tested AD backends
1 parent 2d4b789 commit 462be07

3 files changed

Lines changed: 6 additions & 2 deletions

File tree

test/Project.toml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
1313
Kronecker = "2c470bb0-bcc8-11e8-3dad-c9649493f05e"
1414
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
1515
LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
16+
Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
1617
PDMats = "90014a1f-27ba-587c-ab20-58faa44d9150"
1718
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
1819
ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
@@ -39,4 +40,5 @@ ReverseDiff = "1.2"
3940
SpecialFunctions = "0.10, 1, 2"
4041
StaticArrays = "1"
4142
Statistics = "1"
43+
Mooncake = "0.5"
4244
Zygote = "0.6.38"

test/runtests.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ using ForwardDiff: ForwardDiff
1818
using ReverseDiff: ReverseDiff
1919
using FiniteDifferences: FiniteDifferences
2020
using Compat: only
21-
using ADTypes: AutoForwardDiff, AutoReverseDiff, AutoZygote, AutoFiniteDifferences
21+
using Mooncake: Mooncake
22+
using ADTypes: AutoForwardDiff, AutoReverseDiff, AutoZygote, AutoFiniteDifferences, AutoMooncake, AutoEnzyme
2223
using DifferentiationInterface: DifferentiationInterface as DI
2324
using DifferentiationInterfaceTest: Scenario, test_differentiation
2425

test/test_utils.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ testdiagfunction(k::MOKernel, A, B) = sum(kernelmatrix_diag(k, A, B))
2525
const FD_BACKEND = AutoFiniteDifferences(; fdm=FDM)
2626

2727
const _DEFAULT_BACKENDS = let
28-
backends = [AutoForwardDiff(), AutoReverseDiff()]
28+
backends = [AutoForwardDiff(), AutoReverseDiff(), AutoMooncake()]
2929
_TEST_ZYGOTE && pushfirst!(backends, AutoZygote())
3030
backends
3131
end
@@ -34,6 +34,7 @@ const _BACKEND_MAP = Dict{Symbol,Any}(
3434
:ForwardDiff => AutoForwardDiff(),
3535
:ReverseDiff => AutoReverseDiff(),
3636
:Zygote => AutoZygote(),
37+
:Mooncake => AutoMooncake(),
3738
:FiniteDiff => FD_BACKEND,
3839
)
3940

0 commit comments

Comments
 (0)