Update src/ensemblegpuarray/problem_generation.jl

SciML · Oct 19, 2023 · 5f62f6b · 5f62f6b
1 parent 91e231b
commit 5f62f6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ensemblegpuarray/problem_generation.jl b/src/ensemblegpuarray/problem_generation.jl
@@ -1,4 +1,4 @@
-function generate_problem(prob::AbstractODEProblem, u0, p, jac_prototype, colorvec)
+function generate_problem(prob::SciMLBase.AbstractODEProblem, u0, p, jac_prototype, colorvec)
     _f = let f = prob.f.f, kernel = DiffEqBase.isinplace(prob) ? gpu_kernel : gpu_kernel_oop
         function (du, u, p, t)
             version = get_backend(u)