Remove JET from prereleases as well

Author: ChrisRackauckas

.github/workflows/Tests.yml

@@ -33,7 +33,7 @@ jobs:
           - "LinearSolveHYPRE"
           - "LinearSolvePardiso"
           - "LinearSolveBandedMatrices"
-          - "Enzyme"
+          - "NoPre"
     uses: "SciML/.github/.github/workflows/tests.yml@v1"
     with:
       group: "${{ matrix.group }}"

Project.toml

@@ -84,7 +84,6 @@ GPUArraysCore = "0.1.6, 0.2"
 HYPRE = "1.4.0"
 InteractiveUtils = "1.10"
 IterativeSolvers = "0.9.3"
-JET = "0.8.28, 0.9"
 KernelAbstractions = "0.9.27"
 Krylov = "0.10"
 KrylovKit = "0.8, 0.9"
@@ -130,7 +129,6 @@ ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 HYPRE = "b5ffcf37-a2bd-41ab-a3da-4bd9bc8ad771"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 IterativeSolvers = "42fd0dbc-a981-5370-80f2-aaf504508153"
-JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
 KrylovKit = "0b1a1467-8014-51b9-945f-bf0ae24f4b77"
 KrylovPreconditioners = "45d422c2-293f-44ce-8315-2cb988662dec"
@@ -150,4 +148,4 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [targets]
-test = ["Aqua", "Test", "IterativeSolvers", "InteractiveUtils", "JET", "KrylovKit", "KrylovPreconditioners", "Pkg", "Random", "SafeTestsets", "MultiFloats", "ForwardDiff", "HYPRE", "MPI", "BlockDiagonals", "FiniteDiff", "BandedMatrices", "FastAlmostBandedMatrices", "StaticArrays", "AllocCheck", "StableRNGs", "Zygote", "RecursiveFactorization", "Sparspak", "FastLapackInterface", "SparseArrays"]
+test = ["Aqua", "Test", "IterativeSolvers", "InteractiveUtils", "KrylovKit", "KrylovPreconditioners", "Pkg", "Random", "SafeTestsets", "MultiFloats", "ForwardDiff", "HYPRE", "MPI", "BlockDiagonals", "FiniteDiff", "BandedMatrices", "FastAlmostBandedMatrices", "StaticArrays", "AllocCheck", "StableRNGs", "Zygote", "RecursiveFactorization", "Sparspak", "FastLapackInterface", "SparseArrays"]

test/default_algs.jl

@@ -60,22 +60,6 @@ solve(prob)
 A = rand(4, 4)
 b = rand(4)
 prob = LinearProblem(A, b)
-if if isempty(VERSION.prerelease) 
-      JET.@test_opt init(prob, nothing)
-      JET.@test_opt solve(prob, LUFactorization())
-      JET.@test_opt solve(prob, GenericLUFactorization())
-      @test_skip JET.@test_opt solve(prob, QRFactorization())
-      JET.@test_opt solve(prob, DiagonalFactorization())
-      #JET.@test_opt solve(prob, SVDFactorization())
-      #JET.@test_opt solve(prob, KrylovJL_GMRES())
-
-      prob = LinearProblem(sparse(A), b)
-      #JET.@test_opt solve(prob, UMFPACKFactorization())
-      #JET.@test_opt solve(prob, KLUFactorization())
-      #JET.@test_opt solve(prob, SparspakFactorization())
-      #JET.@test_opt solve(prob)
-end
-
 prob = LinearProblem(sparse(A), b)
 @inferred solve(prob)
 @inferred init(prob, nothing)

test/enzyme/Project.toml renamed to test/nopre/Project.toml

@@ -2,6 +2,7 @@
 FiniteDiff = "6a86dc24-6348-571c-b903-95158fe2bd41"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
+JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LinearSolve = "7ed4a6bd-45f5-4d41-b270-4a48e9bafcae"
 RecursiveFactorization = "f2c3362d-daeb-58d1-803e-2bc74f2840b4"

test/enzyme/enzyme.jl renamed to test/nopre/enzyme.jl

@@ -0,0 +1,16 @@
+using LinearSolve, RecursiveFactorization, LinearAlgebra, SparseArrays, Test
+using JET
+
+JET.@test_opt init(prob, nothing)
+JET.@test_opt solve(prob, LUFactorization())
+JET.@test_opt solve(prob, GenericLUFactorization())
+@test_skip JET.@test_opt solve(prob, QRFactorization())
+JET.@test_opt solve(prob, DiagonalFactorization())
+#JET.@test_opt solve(prob, SVDFactorization())
+#JET.@test_opt solve(prob, KrylovJL_GMRES())
+
+prob = LinearProblem(sparse(A), b)
+#JET.@test_opt solve(prob, UMFPACKFactorization())
+#JET.@test_opt solve(prob, KLUFactorization())
+#JET.@test_opt solve(prob, SparspakFactorization())
+#JET.@test_opt solve(prob)

test/nopre/jet.jl

@@ -22,11 +22,12 @@ if GROUP == "All" || GROUP == "Core"
 end
 
 # Don't run Enzyme tests on prerelease
-if GROUP == "All" || GROUP == "Enzyme" && isempty(VERSION.prerelease)
-    Pkg.activate("enzyme")
+if GROUP == "All" || GROUP == "NoPre" && isempty(VERSION.prerelease)
+    Pkg.activate("nopre")
     Pkg.develop(PackageSpec(path = dirname(@__DIR__)))
     Pkg.instantiate()
-    @time @safetestset "Enzyme Derivative Rules" include("enzyme.jl")
+    @time @safetestset "Enzyme Derivative Rules" include("nopre/enzyme.jl")
+    @time @safetestset "JET Tests" include("nopre/jet.jl")
 end
 
 if GROUP == "DefaultsLoading"

test/runtests.jl

@@ -1,12 +1,16 @@
 using LinearSolve, StaticArrays, LinearAlgebra, Test, StableRNGs
-using AllocCheck
 
 rng = StableRNG(0)
 
 A = SMatrix{5, 5}(Hermitian(rand(rng, 5, 5) + I))
 b = SVector{5}(rand(rng, 5))
 
-@check_allocs __solve_no_alloc(A, b, alg) = solve(LinearProblem(A, b), alg)
+if isempty(VERSION.prerelease)
+    using AllocCheck
+    @check_allocs __solve_no_alloc(A, b, alg) = solve(LinearProblem(A, b), alg)
+else
+    __solve_no_alloc(A, b, alg) = solve(LinearProblem(A, b), alg)
+end
 
 function __non_native_static_array_alg(alg)
     return alg isa SVDFactorization || alg isa KrylovJL