Add ModelingToolkit compatibility test

- Add test for MTK-generated problems with EnsembleGPUKernel
- Adds ModelingToolkit and OrdinaryDiffEqTsit5 test dependencies

Related to #375

Author: ChrisRackauckas

test/Project.toml

@@ -8,15 +8,17 @@ ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+ModelingToolkit = "961ee093-0014-501f-94e3-6117800e7a78"
 Optimization = "7f7a1694-90dd-40f0-9382-eb1efda571ba"
 OptimizationOptimisers = "42dfb2eb-d2b4-4451-abcd-913932933ac1"
 OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
+OrdinaryDiffEqTsit5 = "b1df2697-797e-41e3-8120-5422d3b24e4a"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
+SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 StochasticDiffEq = "789caeaf-c7a9-5a7d-9973-96adeb23e2a0"
-SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 TOML = "fa267f1f-6049-4f14-aa54-33bafae1ed76"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"

test/gpu_kernel_de/mtk_compatibility.jl

@@ -0,0 +1,43 @@
+using DiffEqGPU, Test, ModelingToolkit, OrdinaryDiffEqTsit5, CUDA
+
+@testset "ModelingToolkit GPU Compatibility" begin
+    # Test case from issue #375
+    # ModelingToolkit-generated problems should work with EnsembleGPUKernel
+
+    @parameters σ ρ β
+    @variables t x(t) y(t) z(t)
+    D = Differential(t)
+
+    eqs = [D(x) ~ σ * (y - x),
+        D(y) ~ x * (ρ - z) - y,
+        D(z) ~ x * y - β * z]
+
+    @named sys = ODESystem(eqs, t)
+    sys = structural_simplify(sys)
+
+    u0 = [x => 1.0, y => 0.0, z => 0.0]
+    tspan = (0.0f0, 1.0f0)
+    p = [σ => 10.0, ρ => 28.0, β => 8 / 3]
+
+    prob = ODEProblem(sys, u0, tspan, p)
+
+    # Test that we can create an ensemble problem
+    function prob_func(prob, i, repeat)
+        remake(prob, p = [σ => 10.0 + i * 0.1, ρ => 28.0, β => 8 / 3])
+    end
+
+    ensemble_prob = EnsembleProblem(prob, prob_func = prob_func)
+
+    # This should not error with "CuArray only supports element types that are allocated inline"
+    @test_nowarn begin
+        sol = solve(ensemble_prob, GPUTsit5(), EnsembleGPUKernel(CUDA.CUDABackend());
+            trajectories = 10, dt = 0.1f0, adaptive = false)
+    end
+
+    # Actually test that it works
+    sol = solve(ensemble_prob, GPUTsit5(), EnsembleGPUKernel(CUDA.CUDABackend());
+        trajectories = 10, dt = 0.1f0, adaptive = false)
+
+    @test length(sol) == 10
+    @test all(s.retcode == :Success || s.retcode == :Terminated for s in sol)
+end