test: skip cuda installation + tests on pre-releases

Author: avik-pal

.github/workflows/CI_SimpleNonlinearSolve.yml

@@ -33,11 +33,6 @@ jobs:
           - core
           - adjoint
           - alloc_check
-          - nopre
-        exclude:
-          # Don't run nopre tests on prerelease Julia
-          - group: nopre
-            version: "pre"
     uses: LuxDL/Lux.jl/.github/workflows/CommonCI.yml@ap/ci
     with:
       julia_version: ${{ matrix.version }}
@@ -54,7 +49,6 @@ jobs:
           - core
           - adjoint
           - alloc_check
-          - nopre
     uses: LuxDL/Lux.jl/.github/workflows/CommonCI.yml@ap/ci
     with:
       julia_version: "1.11"

Project.toml

@@ -81,7 +81,6 @@ ArrayInterface = "7.16"
 BandedMatrices = "1.5"
 BenchmarkTools = "1.4"
 BracketingNonlinearSolve = "1"
-CUDA = "5.5"
 CommonSolve = "0.2.4"
 ConcreteStructs = "0.2.3"
 DiffEqBase = "6.158.3"
@@ -138,7 +137,6 @@ julia = "1.10"
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 BandedMatrices = "aae01518-5342-5314-be14-df237901396f"
 BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
-CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 ExplicitImports = "7d51a73a-1435-4ff3-83d9-f097790105c7"
 FastLevenbergMarquardt = "7a0df574-e128-4d35-8cbd-3d84502bf7ce"
 FixedPointAcceleration = "817d07cb-a79a-5c30-9a31-890123675176"
@@ -168,4 +166,4 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [targets]
-test = ["Aqua", "BandedMatrices", "BenchmarkTools", "CUDA", "ExplicitImports", "FastLevenbergMarquardt", "FixedPointAcceleration", "Hwloc", "InteractiveUtils", "LeastSquaresOptim", "LineSearches", "MINPACK", "NLSolvers", "NLsolve", "NaNMath", "NonlinearProblemLibrary", "OrdinaryDiffEqTsit5", "PETSc", "Pkg", "PolyesterForwardDiff", "Random", "ReTestItems", "SIAMFANLEquations", "SparseConnectivityTracer", "SparseMatrixColorings", "SpeedMapping", "StableRNGs", "StaticArrays", "Sundials", "Test", "Zygote"]
+test = ["Aqua", "BandedMatrices", "BenchmarkTools", "ExplicitImports", "FastLevenbergMarquardt", "FixedPointAcceleration", "Hwloc", "InteractiveUtils", "LeastSquaresOptim", "LineSearches", "MINPACK", "NLSolvers", "NLsolve", "NaNMath", "NonlinearProblemLibrary", "OrdinaryDiffEqTsit5", "PETSc", "Pkg", "PolyesterForwardDiff", "Random", "ReTestItems", "SIAMFANLEquations", "SparseConnectivityTracer", "SparseMatrixColorings", "SpeedMapping", "StableRNGs", "StaticArrays", "Sundials", "Test", "Zygote"]

test/cuda_tests.jl

@@ -1,4 +1,4 @@
-@testitem "CUDA Tests" tags=[:cuda] begin
+@testitem "CUDA Tests" tags=[:cuda] skip=:(!isempty(VERSION.prerelease)) begin
     using CUDA, NonlinearSolve, LinearSolve, StableRNGs
 
     if CUDA.functional()
@@ -44,7 +44,7 @@
     end
 end
 
-@testitem "Termination Conditions: Allocations" tags=[:cuda] begin
+@testitem "Termination Conditions: Allocations" tags=[:cuda] skip=:(!isempty(VERSION.prerelease)) begin
     using CUDA, NonlinearSolveBase, Test, LinearAlgebra
 
     if CUDA.functional()
@@ -67,8 +67,9 @@ end
             end
 
             @testset "Mode: $(tcond)" for tcond in NORM_TERMINATION_CONDITIONS
-                for nfn in
-                    (Base.Fix1(maximum, abs), Base.Fix2(norm, 2), Base.Fix2(norm, Inf))
+                for nfn in (
+                    Base.Fix1(maximum, abs), Base.Fix2(norm, 2), Base.Fix2(norm, Inf)
+                )
                     @test_nowarn NonlinearSolveBase.check_convergence(
                         tcond(nfn), du, u, uprev, 1e-3, 1e-3)
                 end

test/runtests.jl

@@ -35,6 +35,12 @@ if GROUP == "all" || GROUP == "nopre"
         push!(EXTRA_PKGS, Pkg.PackageSpec("Enzyme"))
     end
 end
+if GROUP == "all" || GROUP == "cuda"
+    # Only add CUDA for cuda group if not on prerelease Julia
+    if isempty(VERSION.prerelease)
+        push!(EXTRA_PKGS, Pkg.PackageSpec("CUDA"))
+    end
+end
 length(EXTRA_PKGS) ≥ 1 && Pkg.add(EXTRA_PKGS)
 
 # Use sequential execution for wrapper tests to avoid parallel initialization issues