Implement DifferentiationInterfaceTest infrastructure

Author: simsurace

test/Project.toml

@@ -1,8 +1,11 @@
 [deps]
+ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 AxisArrays = "39de3d68-74b9-583c-8d2d-e117c070f3a9"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
+DifferentiationInterface = "a0c0ee7d-e4b9-4e03-894e-1c5f64a51d63"
+DifferentiationInterfaceTest = "a82114a7-5aa3-49a8-9643-716bb13727a3"
 Distances = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7"
 Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
 FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
@@ -22,8 +25,11 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
+ADTypes = "1"
 AxisArrays = "0.4.3"
 Compat = "3, 4"
+DifferentiationInterface = "0.7"
+DifferentiationInterfaceTest = "0.11"
 Distances = "0.10"
 Documenter = "0.25, 0.26, 0.27, 1"
 FiniteDifferences = "0.10.8, 0.11, 0.12"

test/basekernels/exponential.jl

@@ -22,6 +22,7 @@
         # Standardised tests.
         TestUtils.test_interface(k)
         test_ADs(SEKernel)
+        test_ADs_DIT(SEKernel)
         test_interface_ad_perf(_ -> SEKernel(), nothing, StableRNG(123456))
     end
     @testset "ExponentialKernel" begin
@@ -41,6 +42,7 @@
         # Standardised tests.
         TestUtils.test_interface(k)
         test_ADs(ExponentialKernel)
+        test_ADs_DIT(ExponentialKernel)
         test_interface_ad_perf(_ -> ExponentialKernel(), nothing, StableRNG(123456))
     end
     @testset "GammaExponentialKernel" begin
@@ -59,6 +61,7 @@
         @test k2(v1, v2) ≈ k(v1, v2)
 
         test_ADs(γ -> GammaExponentialKernel(; gamma=only(γ)), [1 + 0.5 * rand()])
+        test_ADs_DIT(γ -> GammaExponentialKernel(; gamma=only(γ)), [1 + 0.5 * rand()])
         test_params(k, ([γ],))
         TestUtils.test_interface(GammaExponentialKernel(; γ=1.36))
 

test/basekernels/fbm.jl

@@ -14,6 +14,7 @@
     test_interface(k; rtol=1e-5)
     @test repr(k) == "Fractional Brownian Motion Kernel (h = $(h))"
     test_ADs(FBMKernel; ADs=[:ReverseDiff])
+    test_ADs_DIT(FBMKernel; ADs=[:ReverseDiff])
 
     # Tests failing for ForwardDiff and Zygote@0.6.
     # Related to: https://github.com/FluxML/Zygote.jl/issues/1036

test/basekernels/matern.jl

@@ -21,6 +21,7 @@
         # Standardised tests.
         TestUtils.test_interface(k, Float64)
         test_ADs(() -> MaternKernel(; nu=ν))
+        test_ADs_DIT(() -> MaternKernel(; nu=ν))
 
         test_params(k, ([ν],))
 
@@ -60,6 +61,7 @@
         # Standardised tests.
         TestUtils.test_interface(k, Float64)
         test_ADs(Matern32Kernel)
+        test_ADs_DIT(Matern32Kernel)
         test_interface_ad_perf(_ -> Matern32Kernel(), nothing, StableRNG(123456))
     end
     @testset "Matern52Kernel" begin

test/kernels/scaledkernel.jl

@@ -12,6 +12,7 @@
     # Standardised tests.
     TestUtils.test_interface(ks, Float64)
     test_ADs(x -> exp(x[1]) * SqExponentialKernel(), rand(1))
+    test_ADs_DIT(x -> exp(x[1]) * SqExponentialKernel(), rand(1))
     test_interface_ad_perf(c -> c * SEKernel(), 0.3, StableRNG(123456))
 
     test_params(s * k, (k, [s]))

test/mokernels/intrinsiccoregion.jl

@@ -41,6 +41,7 @@
     )
 
     test_ADs(icoregionkernel; dims=dims)
+    test_ADs_DIT(icoregionkernel; dims=dims)
 
     @test string(icoregionkernel) ==
         string("Intrinsic Coregion Kernel: ", kernel, " with ", dims.out, " outputs")

test/runtests.jl

@@ -20,6 +20,9 @@ using ForwardDiff: ForwardDiff
 using ReverseDiff: ReverseDiff
 using FiniteDifferences: FiniteDifferences
 using Compat: only
+using ADTypes: AutoForwardDiff, AutoReverseDiff, AutoZygote, AutoFiniteDifferences
+using DifferentiationInterface: DifferentiationInterface as DI
+using DifferentiationInterfaceTest: Scenario, test_differentiation
 
 using KernelFunctions: SimpleKernel, metric, kappa, ColVecs, RowVecs, TestUtils