Merge pull request #2508 from jClugstor/ADTypesSwitch

Switch to using ADTypes for AD choice
SciML · Dec 21, 2024 · 3921078 · 3921078
2 parents 8353139 + 0bd7013
commit 3921078
Show file tree

Hide file tree

Showing 65 changed files with 889 additions and 511 deletions.
diff --git a/lib/OrdinaryDiffEqAdamsBashforthMoulton/src/OrdinaryDiffEqAdamsBashforthMoulton.jl b/lib/OrdinaryDiffEqAdamsBashforthMoulton/src/OrdinaryDiffEqAdamsBashforthMoulton.jl
@@ -9,12 +9,13 @@ import OrdinaryDiffEqCore: OrdinaryDiffEqMutableCache, OrdinaryDiffEqConstantCac
                            constvalue, calculate_residuals, calculate_residuals!,
                            trivial_limiter!, get_fsalfirstlast,
                            generic_solver_docstring,
-                           full_cache
+                           full_cache,
+                           _bool_to_ADType
 import OrdinaryDiffEqLowOrderRK: BS3ConstantCache, BS3Cache, RK4ConstantCache, RK4Cache
 import RecursiveArrayTools: recursivefill!
 using MuladdMacro, FastBroadcast
 import Static: False
-import ADTypes: AutoForwardDiff
+import ADTypes: AutoForwardDiff, AbstractADType
 import OrdinaryDiffEqCore
 
 using Reexport

diff --git a/lib/OrdinaryDiffEqBDF/Project.toml b/lib/OrdinaryDiffEqBDF/Project.toml
@@ -1,9 +1,10 @@
 name = "OrdinaryDiffEqBDF"
 uuid = "6ad6398a-0878-4a85-9266-38940aa047c8"
 authors = ["ParamThakkar123 <[email protected]>"]
-version = "1.1.2"
+version = "1.2.0"
 
 [deps]
+ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 FastBroadcast = "7034ab61-46d4-4ed7-9d0f-46aef9175898"
@@ -31,7 +32,7 @@ LinearAlgebra = "<0.0.1, 1"
 MacroTools = "0.5.13"
 MuladdMacro = "0.2.4"
 ODEProblemLibrary = "0.1.8"
-OrdinaryDiffEqCore = "1.1"
+OrdinaryDiffEqCore = "1.11"
 OrdinaryDiffEqDifferentiation = "<0.0.1, 1"
 OrdinaryDiffEqNonlinearSolve = "<0.0.1, 1"
 OrdinaryDiffEqSDIRK = "<0.0.1, 1"

diff --git a/lib/OrdinaryDiffEqBDF/src/OrdinaryDiffEqBDF.jl b/lib/OrdinaryDiffEqBDF/src/OrdinaryDiffEqBDF.jl
@@ -20,7 +20,8 @@ import OrdinaryDiffEqCore: alg_order, calculate_residuals!,
                            step_accept_controller!,
                            step_reject_controller!, post_newton_controller!,
                            u_modified!, DAEAlgorithm, _unwrap_val, DummyController,
-                           get_fsalfirstlast, generic_solver_docstring
+                           get_fsalfirstlast, generic_solver_docstring, _bool_to_ADType,
+                           _process_AD_choice
 using OrdinaryDiffEqSDIRK: ImplicitEulerConstantCache, ImplicitEulerCache
 
 using TruncatedStacktraces, MuladdMacro, MacroTools, FastBroadcast, RecursiveArrayTools
@@ -32,6 +33,8 @@ using OrdinaryDiffEqDifferentiation: UJacobianWrapper
 using OrdinaryDiffEqNonlinearSolve: NLNewton, du_alias_or_new, build_nlsolver,
                                     nlsolve!, nlsolvefail, isnewton, markfirststage!,
                                     set_new_W!, DIRK, compute_step!, COEFFICIENT_MULTISTEP
+import ADTypes
+import ADTypes: AutoForwardDiff, AutoFiniteDiff, AbstractADType
 
 using Reexport
 @reexport using DiffEqBase