Merge pull request #388 from ChrisRackauckas-Claude/static-improvements-20251229-211729

Add JET.jl static analysis tests

Author: ChrisRackauckas

test/Project.toml

@@ -6,6 +6,7 @@ DiffEqDevTools = "f3b72e0c-5b89-59e1-b016-84e28bfd966d"
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
+JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 OpenCL = "08131aa3-fb12-5dee-8b74-c09406e224a2"

test/jet_tests.jl

@@ -0,0 +1,44 @@
+using Test
+using JET
+using DiffEqGPU
+using StaticArrays
+
+@testset "JET static analysis" begin
+    # Test that algorithm constructors are type-stable
+    @testset "Algorithm constructors" begin
+        # Explicit algorithms
+        @test_opt GPUTsit5()
+        @test_opt GPUVern7()
+        @test_opt GPUVern9()
+
+        # SDE algorithms
+        @test_opt GPUEM()
+        @test_opt GPUSIEA()
+
+        # Implicit algorithms (with default AD)
+        @test_opt GPURosenbrock23()
+        @test_opt GPURodas4()
+        @test_opt GPURodas5P()
+        @test_opt GPUKvaerno3()
+        @test_opt GPUKvaerno5()
+    end
+
+    @testset "Ensemble algorithm constructors" begin
+        @test_opt EnsembleCPUArray()
+    end
+
+    @testset "alg_order type stability" begin
+        @test_opt DiffEqGPU.alg_order(GPUTsit5())
+        @test_opt DiffEqGPU.alg_order(GPUVern7())
+        @test_opt DiffEqGPU.alg_order(GPUVern9())
+        @test_opt DiffEqGPU.alg_order(GPUEM())
+        @test_opt DiffEqGPU.alg_order(GPUSIEA())
+    end
+
+    @testset "Utility functions type stability" begin
+        # Test diffeqgpunorm with different input types
+        @test_opt DiffEqGPU.diffeqgpunorm([1.0f0, 2.0f0, 3.0f0], 0.0f0)
+        @test_opt DiffEqGPU.diffeqgpunorm(1.0f0, 0.0f0)
+        @test_opt DiffEqGPU.diffeqgpunorm(SA[1.0f0, 2.0f0, 3.0f0], 0.0f0)
+    end
+end

test/runtests.jl

@@ -19,6 +19,10 @@ const GROUP = get(ENV, "GROUP", "CUDA")
 
 using SafeTestsets, Test
 
+@time @safetestset "JET static analysis" begin
+    include("jet_tests.jl")
+end
+
 @time @safetestset "GPU Kernelized Stiff ODE Mass Matrix" begin
     include("gpu_kernel_de/stiff_ode/gpu_ode_mass_matrix.jl")
 end