Merge pull request #1113 from SciML/compathelper/new_version/2024-09-17-22-09-03-606-02207253346

CompatHelper: bump compat for DiffEqFlux to 4 for package gpu, (keep existing compat)

Author: ChrisRackauckas

test/gpu/Project.toml

@@ -7,5 +7,5 @@ LuxCUDA = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 [compat]
 CUDA = "3.12, 4, 5"
 DiffEqCallbacks = "2.24, 3"
-DiffEqFlux = "3"
+DiffEqFlux = "3, 4"
 LuxCUDA = "0.3.1"

test/gpu/mixed_gpu_cpu_adjoint.jl

@@ -1,6 +1,8 @@
 using SciMLSensitivity, OrdinaryDiffEq
 using Lux, LuxCUDA, Test, Zygote, Random, LinearAlgebra, ComponentArrays
 
+const gdev = gpu_device()
+
 CUDA.allowscalar(false)
 
 H = CuArray(rand(Float32, 2, 2))
@@ -42,10 +44,10 @@ grad = Zygote.gradient(cost, p)[1]
 rng = MersenneTwister(1234)
 m = 32
 n = 16
-Z = randn(rng, Float32, (n, m)) |> gpu
+Z = randn(rng, Float32, (n, m)) |> gdev
 𝒯 = 2.0f0
 Δτ = 1.0f-1
-ca_init = [zeros(1); ones(m)] |> gpu
+ca_init = [zeros(1); ones(m)] |> gdev
 
 function f(ca, Z, t)
     a = ca[2:end]
@@ -54,7 +56,7 @@ function f(ca, Z, t)
     Ka_unit = Z' * w_unit
     z_unit = dot(abs.(Ka_unit), a_unit)
     aKa_over_z = a .* Ka_unit / z_unit
-    [sum(aKa_over_z) / m; -abs.(aKa_over_z)] |> gpu
+    [sum(aKa_over_z) / m; -abs.(aKa_over_z)] |> gdev
 end
 
 function c(Z)