Fix warnings

Author: ErikQQY

Project.toml

@@ -18,6 +18,7 @@ PreallocationToolsSparseConnectivityTracerExt = "SparseConnectivityTracer"
 
 [compat]
 Adapt = "4.1.1"
+ADTypes = "1.13"
 Aqua = "0.8.11"
 ArrayInterface = "7.18.0"
 ForwardDiff = "0.10.38"
@@ -38,6 +39,7 @@ Test = "1.10"
 julia = "1.10"
 
 [extras]
+ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 LabelledArrays = "2ee39098-c373-598a-b85f-a56591580800"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -55,4 +57,4 @@ Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Aqua", "Random", "LabelledArrays", "LinearAlgebra", "OrdinaryDiffEq", "Test", "RecursiveArrayTools", "Pkg", "SafeTestsets", "Optimization", "OptimizationOptimJL", "SparseArrays", "Symbolics", "SparseConnectivityTracer"]
+test = ["Aqua", "ADTypes", "Random", "LabelledArrays", "LinearAlgebra", "OrdinaryDiffEq", "Test", "RecursiveArrayTools", "Pkg", "SafeTestsets", "Optimization", "OptimizationOptimJL", "SparseArrays", "Symbolics", "SparseConnectivityTracer"]

test/core_nesteddual.jl

@@ -90,11 +90,13 @@ newtonsol = solve(optprob, Newton())
 cache = DiffCache(zeros(ps, ps), [4, 4, 2])
 prob = ODEProblem{true, SciMLBase.FullSpecialize}(foo, ones(ps, ps), (0.0, 1.0),
     (coeffs, cache))
-realsol = solve(prob, TRBDF2(chunk_size = 2), saveat = 0.0:0.1:10.0, reltol = 1e-8)
+realsol = solve(prob, TRBDF2(autodiff = AutoForwardDiff(chunksize = 2)),
+    saveat = 0.0:0.1:10.0, reltol = 1e-8)
 
 function objfun(x, prob, realsol, cache)
     prob = remake(prob, u0 = eltype(x).(prob.u0), p = (x, cache))
-    sol = solve(prob, TRBDF2(chunk_size = 2), saveat = 0.0:0.1:10.0, reltol = 1e-8)
+    sol = solve(prob, TRBDF2(autodiff = AutoForwardDiff(chunksize = 2)),
+        saveat = 0.0:0.1:10.0, reltol = 1e-8)
 
     ofv = 0.0
     if any((s.retcode != ReturnCode.Success for s in sol))

test/core_odes.jl

@@ -1,6 +1,6 @@
 using LinearAlgebra,
       OrdinaryDiffEq, Test, PreallocationTools, LabelledArrays,
-      RecursiveArrayTools
+      RecursiveArrayTools, ADTypes
 
 #Base array
 function foo(du, u, (A, tmp), t)
@@ -15,12 +15,12 @@ u0 = ones(5, 5)
 A = ones(5, 5)
 cache = DiffCache(zeros(5, 5), chunk_size)
 prob = ODEProblem{true, SciMLBase.FullSpecialize}(foo, u0, (0.0, 1.0), (A, cache))
-sol = solve(prob, Rodas5P(chunk_size = chunk_size))
+sol = solve(prob, Rodas5P(autodiff = AutoForwardDiff(chunksize = chunk_size)))
 @test sol.retcode == ReturnCode.Success
 
 cache = FixedSizeDiffCache(zeros(5, 5), chunk_size)
 prob = ODEProblem{true, SciMLBase.FullSpecialize}(foo, u0, (0.0, 1.0), (A, cache))
-sol = solve(prob, Rodas5P(chunk_size = chunk_size))
+sol = solve(prob, Rodas5P(autodiff = AutoForwardDiff(chunksize = chunk_size)))
 @test sol.retcode == ReturnCode.Success
 
 #with auto-detected chunk_size
@@ -60,7 +60,7 @@ end
 chunk_size = 4
 prob = ODEProblem{true, SciMLBase.FullSpecialize}(foo, u0, (0.0, 1.0),
     (A, DiffCache(c, chunk_size)))
-sol = solve(prob, Rodas5P(chunk_size = chunk_size))
+sol = solve(prob, Rodas5P(autodiff = AutoForwardDiff(chunksize = chunk_size)))
 @test sol.retcode == ReturnCode.Success
 #with auto-detected chunk_size
 prob = ODEProblem{true, SciMLBase.FullSpecialize}(foo, u0, (0.0, 1.0), (A, DiffCache(c)))

test/gpu_all.jl

@@ -61,12 +61,12 @@ u0 = cu(rand(10, 10)) #example kept small for test purposes.
 A = cu(-randn(10, 10))
 cache = DiffCache(cu(zeros(10, 10)), chunk_size)
 prob = ODEProblem{true, SciMLBase.FullSpecialize}(foo, u0, (0.0f0, 1.0f0), (A, cache))
-sol = solve(prob, TRBDF2(chunk_size = chunk_size))
+sol = solve(prob, TRBDF2(autodiff = AutoForwardDiff(chunksize = chunk_size)))
 @test sol.retcode == ReturnCode.Success
 
 cache = FixedSizeDiffCache(cu(zeros(10, 10)), chunk_size)
 prob = ODEProblem{true, SciMLBase.FullSpecialize}(foo, u0, (0.0f0, 1.0f0), (A, cache))
-sol = solve(prob, TRBDF2(chunk_size = chunk_size))
+sol = solve(prob, TRBDF2(autodiff = AutoForwardDiff(chunksize = chunk_size)))
 @test sol.retcode == ReturnCode.Success
 
 #with auto-detected chunk_size