Skip type stability on 1.12

Author: gdalle

.github/workflows/Test.yml

@@ -63,6 +63,12 @@ jobs:
             version: "pre"
           - version: "1"
             group: Back/ChainRules
+          - version: "pre"
+            group: Back/Enzyme
+          - version: "pre"
+            group: Back/Mooncake
+          - version: "pre"
+            group: Back/DifferentiateWith
     env:
       JULIA_DI_TEST_GROUP: ${{ matrix.group }}
       JULIA_DI_PR_DRAFT: ${{ github.event.pull_request.draft }}

DifferentiationInterface/test/Back/Enzyme/test.jl

@@ -74,19 +74,6 @@ end;
     )
 end
 
-#=
-# TODO: reactivate type stability tests
-
-test_differentiation(
-    AutoEnzyme(; mode=Enzyme.Forward),  # TODO: add more
-    default_scenarios(; include_batchified=false);
-    correctness=false,
-    type_stability=:prepared,
-    excluded=SECOND_ORDER,
-    logging=LOGGING,
-);
-=#
-
 @testset "Second order" begin
     test_differentiation(
         [

DifferentiationInterface/test/Back/ForwardDiff/test.jl

@@ -48,13 +48,16 @@ end
     )
 
     test_differentiation(
-        AutoForwardDiff(); correctness=false, type_stability=:prepared, logging=LOGGING
+        AutoForwardDiff();
+        correctness=false,
+        type_stability=safetypestab(:prepared),
+        logging=LOGGING,
     )
 
     test_differentiation(
         AutoForwardDiff(; chunksize=5);
         correctness=false,
-        type_stability=:full,
+        type_stability=safetypestab(:full),
         excluded=[:hessian],
         logging=LOGGING,
     )

DifferentiationInterface/test/Back/GTPSA/test.jl

@@ -19,16 +19,16 @@ end
 test_differentiation(
     AutoGTPSA(),
     default_scenarios(; include_constantified=true);
-    type_stability=:full,
+    type_stability=safetypestab(:full),
     logging=LOGGING,
 );
 
 # Test with Descriptor:
 d1 = GTPSA.Descriptor(20, 2) # 20 variables to 2nd order
-test_differentiation(AutoGTPSA(d1); type_stability=:full, logging=LOGGING);
+test_differentiation(AutoGTPSA(d1); type_stability=safetypestab(:full), logging=LOGGING);
 
 # Test with Descriptor using varying orders
 vos = 2 * ones(Int, 20)
 vos[1] = 3
 d2 = GTPSA.Descriptor(vos, 3)
-test_differentiation(AutoGTPSA(d2); type_stability=:full, logging=LOGGING);
+test_differentiation(AutoGTPSA(d2); type_stability=safetypestab(:full), logging=LOGGING);

DifferentiationInterface/test/Core/ZeroBackends/test.jl

@@ -27,15 +27,15 @@ end
         ],
         default_scenarios(; include_batchified=false, include_constantified=true);
         correctness=false,
-        type_stability=:full,
+        type_stability=safetypestab(:full),
         logging=LOGGING,
     )
 
     test_differentiation(
         AutoSparse.(zero_backends, coloring_algorithm=GreedyColoringAlgorithm()),
         default_scenarios(; include_constantified=true);
         correctness=false,
-        type_stability=:full,
+        type_stability=safetypestab(:full),
         excluded=[
             :pushforward, :pullback, :gradient, :derivative, :hvp, :second_derivative
         ],

DifferentiationInterface/test/testutils.jl

@@ -20,3 +20,5 @@ function MyAutoSparse(backend::AbstractADType)
         coloring_algorithm=GreedyColoringAlgorithm(),
     )
 end
+
+safetypestab(symb) = VERSION < v"1.12-" ? :none : symb  # TODO: remove

DifferentiationInterfaceTest/CHANGELOG.md

@@ -5,7 +5,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [Unreleased](https://github.com/JuliaDiff/DifferentiationInterface.jl/compare/DifferentiationInterfaceTest-v0.10.1...main)
+## [Unreleased](https://github.com/JuliaDiff/DifferentiationInterface.jl/compare/DifferentiationInterfaceTest-v0.10.2...main)
+
+### Fixed
+
+  - Allow JET v0.10 but disable test stability
+
+## [0.10.2](https://github.com/JuliaDiff/DifferentiationInterface.jl/compare/DifferentiationInterfaceTest-v0.10.1...DifferentiationInterfaceTest-v0.10.2)
 
 ### Fixed
 

DifferentiationInterfaceTest/test/formalities.jl

@@ -10,7 +10,10 @@ using Test
     Aqua.test_all(DifferentiationInterfaceTest; ambiguities=false, undocumented_names=true)
 end
 @testset verbose = true "JET" begin
-    JET.test_package(DifferentiationInterfaceTest; target_defined_modules=true)
+    # until https://github.com/JuliaLang/julia/pull/59321 is released
+    if VERSION <= v"1.12-"
+        JET.test_package(DifferentiationInterfaceTest; target_defined_modules=true)
+    end
 end
 
 @testset "Documentation" begin

DifferentiationInterfaceTest/test/runtests.jl

@@ -17,6 +17,8 @@ using DifferentiationInterfaceTest:
 
 GROUP = get(ENV, "JULIA_DIT_TEST_GROUP", "All")
 
+safetypestab(symb) = VERSION < v"1.12-" ? :none : symb  # TODO: remove
+
 ## Main tests
 
 @time @testset verbose = true "DifferentiationInterfaceTest.jl" begin

DifferentiationInterfaceTest/test/zero_backends.jl

@@ -12,7 +12,7 @@ LOGGING = get(ENV, "CI", "false") == "false"
 test_differentiation(
     AutoZeroForward(),
     map(zero, default_scenarios(; include_batchified=false));
-    type_stability=:full,
+    type_stability=safetypestab(:full),
     logging=LOGGING,
     reprepare=false,
 )
@@ -24,7 +24,7 @@ test_differentiation(
         default_scenarios(; include_batchified=false),
     );
     correctness=false,
-    type_stability=:prepared,
+    type_stability=safetypestab(:prepared),
     logging=LOGGING,
 )