Merge pull request #351 from SciML/immutable

Use the ImmutableODEProblem from SciMLBase

Author: ChrisRackauckas

Project.toml

@@ -52,7 +52,7 @@ MuladdMacro = "0.2"
 Parameters = "0.12"
 RecursiveArrayTools = "2, 3"
 Requires = "1.0"
-SciMLBase = "2.86.1"
+SciMLBase = "2.92"
 Setfield = "1"
 SimpleDiffEq = "1"
 StaticArrays = "1"

src/DiffEqGPU.jl

@@ -23,6 +23,7 @@ using ForwardDiff
 import StaticArrays: StaticVecOrMat, @_inline_meta
 # import LinearAlgebra: \
 import StaticArrays: LU, StaticLUMatrix, arithmetic_closure
+import SciMLBase: ImmutableODEProblem
 
 abstract type EnsembleArrayAlgorithm <: SciMLBase.EnsembleAlgorithm end
 abstract type EnsembleKernelAlgorithm <: SciMLBase.EnsembleAlgorithm end
@@ -37,7 +38,6 @@ include("ensemblegpuarray/kernels.jl")
 include("ensemblegpuarray/problem_generation.jl")
 include("ensemblegpuarray/lowerlevel_solve.jl")
 
-include("ensemblegpukernel/problems/ode_problems.jl")
 include("ensemblegpukernel/callbacks.jl")
 include("ensemblegpukernel/lowerlevel_solve.jl")
 include("ensemblegpukernel/gpukernel_algorithms.jl")

src/ensemblegpukernel/integrators/nonstiff/types.jl

@@ -1,24 +1,5 @@
 ## Fixed TimeStep Integrator
 
-function Adapt.adapt_structure(to, prob::Union{ODEProblem{<:Any, <:Any, iip}, ImmutableODEProblem{<:Any, <:Any, iip}}) where {iip}
-    ImmutableODEProblem{iip, true}(adapt(to, prob.f),
-        adapt(to, prob.u0),
-        adapt(to, prob.tspan),
-        adapt(to, prob.p);
-        adapt(to, prob.kwargs)...)
-end
-
-function Adapt.adapt_structure(to, f::ODEFunction{iip}) where {iip}
-    if f.mass_matrix !== I && f.initialization_data !== nothing
-        error("Adaptation to GPU failed: DAEs of ModelingToolkit currently not supported.")
-    end
-    ODEFunction{iip, SciMLBase.FullSpecialize}(f.f, jac = f.jac, mass_matrix = f.mass_matrix,
-    initializeprobmap = f.initializeprobmap, 
-    initializeprobpmap = f.initializeprobpmap, 
-    update_initializeprob! = f.update_initializeprob!, 
-    initialization_data = nothing, initializeprob = nothing)
-end
-
 mutable struct GPUTsit5Integrator{IIP, S, T, ST, P, F, TS, CB, AlgType} <:
                DiffEqBase.AbstractODEIntegrator{AlgType, IIP, S, T}
     alg::AlgType