Improve JET tests to cover all solvers

jet_test_results.txt

@@ -0,0 +1,26 @@
+Test Summary:                      | Pass  Broken  Total   Time
+JET Tests for Dense Factorizations |    9       4     13  18.3s
+JET Tests for Extension Factorizations: Error During Test at /home/crackauc/sandbox/tmp_20250809_074544_96568/LinearSolve.jl/test/nopre/jet.jl:56
+ Unexpected Pass
+ Expression: #= /home/crackauc/sandbox/tmp_20250809_074544_96568/LinearSolve.jl/test/nopre/jet.jl:56 =# JET.@test_opt solve(prob, MKLLUFactorization())
+ Got correct result, please change to @test if no longer broken.
+
+JET Tests for Extension Factorizations: Error During Test at /home/crackauc/sandbox/tmp_20250809_074544_96568/LinearSolve.jl/test/nopre/jet.jl:65
+  Test threw exception
+  Expression: #= /home/crackauc/sandbox/tmp_20250809_074544_96568/LinearSolve.jl/test/nopre/jet.jl:65 =# JET.@test_opt solve(prob, CudaOffloadFactorization())
+  CudaOffloadFactorization requires that CUDA is loaded, i.e. `using CUDA`
+  Stacktrace:
+   [1] error(s::String)
+     @ Base ./error.jl:35
+   [2] CudaOffloadFactorization
+     @ ~/sandbox/tmp_20250809_074544_96568/LinearSolve.jl/src/extension_algs.jl:78 [inlined]
+   [3] macro expansion
+     @ ~/.julia/packages/JET/yNWjn/src/JETBase.jl:1004 [inlined]
+   [4] macro expansion
+     @ ~/sandbox/tmp_20250809_074544_96568/LinearSolve.jl/test/nopre/jet.jl:65 [inlined]
+   [5] macro expansion
+     @ ~/.julia/juliaup/julia-1.11.6+0.x64.linux.gnu/share/julia/stdlib/v1.11/Test/src/Test.jl:1709 [inlined]
+   [6] top-level scope
+     @ ~/sandbox/tmp_20250809_074544_96568/LinearSolve.jl/test/nopre/jet.jl:48
+Test Summary:                          | Pass  Error  Broken  Total  Time
+JET Tests for Extension Factorizations |    1      2       3      6  9.9s

test/nopre/jet.jl

@@ -1,19 +1,110 @@
 using LinearSolve, RecursiveFactorization, LinearAlgebra, SparseArrays, Test
 using JET
 
+# Dense problem setup
 A = rand(4, 4)
 b = rand(4)
 prob = LinearProblem(A, b)
-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)
+
+# Symmetric positive definite matrix for Cholesky
+A_spd = A' * A + I
+prob_spd = LinearProblem(A_spd, b)
+
+# Symmetric matrix for LDLt
+A_sym = A + A'
+prob_sym = LinearProblem(A_sym, b)
+
+# Sparse problem setup
+A_sparse = sparse(A)
+prob_sparse = LinearProblem(A_sparse, b)
+
+# Sparse SPD for CHOLMODFactorization
+A_sparse_spd = sparse(A_spd)
+prob_sparse_spd = LinearProblem(A_sparse_spd, b)
+
+@testset "JET Tests for Dense Factorizations" begin
+    # Working tests - these pass JET optimization checks
+    JET.@test_opt init(prob, nothing)
+    JET.@test_opt solve(prob, LUFactorization())
+    JET.@test_opt solve(prob, GenericLUFactorization())
+    JET.@test_opt solve(prob, DiagonalFactorization())
+    JET.@test_opt solve(prob, SimpleLUFactorization())
+    JET.@test_opt solve(prob_spd, NormalCholeskyFactorization())
+    JET.@test_opt solve(prob, NormalBunchKaufmanFactorization())
+
+    # CholeskyFactorization and SVDFactorization now pass JET tests
+    JET.@test_opt solve(prob_spd, CholeskyFactorization())
+    JET.@test_opt solve(prob, SVDFactorization())
+
+    # Tests with known type stability issues - marked as broken
+    JET.@test_opt solve(prob, QRFactorization()) broken=true
+    JET.@test_opt solve(prob_sym, LDLtFactorization()) broken=true
+    JET.@test_opt solve(prob_sym, BunchKaufmanFactorization()) broken=true
+    JET.@test_opt solve(prob, GenericFactorization()) broken=true
+end
+
+@testset "JET Tests for Extension Factorizations" begin
+    # RecursiveFactorization.jl extensions
+    JET.@test_opt solve(prob, RFLUFactorization())
+
+    # Tests with known type stability issues
+    JET.@test_opt solve(prob, FastLUFactorization()) broken=true
+    JET.@test_opt solve(prob, FastQRFactorization()) broken=true
+
+    # Platform-specific factorizations (may not be available on all systems)
+    if @isdefined(MKLLUFactorization)
+        # MKLLUFactorization passes JET tests
+        JET.@test_opt solve(prob, MKLLUFactorization())
+    end
+
+    if Sys.isapple() && @isdefined(AppleAccelerateLUFactorization)
+        JET.@test_opt solve(prob, AppleAccelerateLUFactorization()) broken=true
+    end
+
+    # CUDA/Metal factorizations (only test if CUDA/Metal are loaded)
+    # CudaOffloadFactorization requires CUDA to be loaded, skip if not available
+    if @isdefined(MetalLUFactorization)
+        JET.@test_opt solve(prob, MetalLUFactorization()) broken=true
+    end
+    if @isdefined(BLISLUFactorization)
+        JET.@test_opt solve(prob, BLISLUFactorization()) broken=true
+    end
+end
+
+@testset "JET Tests for Sparse Factorizations" begin
+    JET.@test_opt solve(prob_sparse, UMFPACKFactorization()) broken=true
+    JET.@test_opt solve(prob_sparse, KLUFactorization()) broken=true
+    JET.@test_opt solve(prob_sparse_spd, CHOLMODFactorization()) broken=true
+
+    # SparspakFactorization requires Sparspak to be loaded
+    # PardisoJL requires Pardiso to be loaded
+    # CUSOLVERRFFactorization requires CUSOLVERRF to be loaded
+    # These are tested in their respective extension test suites
+end
+
+@testset "JET Tests for Krylov Methods" begin
+    # KrylovJL methods that pass JET tests
+    JET.@test_opt solve(prob_spd, KrylovJL_CG())
+    JET.@test_opt solve(prob, KrylovJL_BICGSTAB())
+    JET.@test_opt solve(prob, KrylovJL_LSMR())
+    JET.@test_opt solve(prob, KrylovJL_CRAIGMR())
+
+    # SimpleGMRES passes JET tests
+    JET.@test_opt solve(prob, SimpleGMRES())
+
+    # KrylovJL methods with known type stability issues
+    JET.@test_opt solve(prob, KrylovJL_GMRES()) broken=true
+    JET.@test_opt solve(prob_sym, KrylovJL_MINRES()) broken=true
+    JET.@test_opt solve(prob_sym, KrylovJL_MINARES()) broken=true
+
+    # Extension Krylov methods (require extensions)
+    # KrylovKitJL_CG, KrylovKitJL_GMRES require KrylovKit to be loaded
+    # IterativeSolversJL requires IterativeSolvers to be loaded
+    # These are tested in their respective extension test suites
+end
+
+@testset "JET Tests for Default Solver" begin
+    # Test the default solver selection
+    JET.@test_opt solve(prob) broken=true
+    JET.@test_opt solve(prob_sparse) broken=true
+end