Merge pull request #2638 from SciML/ChrisRackauckas-patch-2

Test whether all latest releases together means tests pass

Author: ChrisRackauckas

Project.toml

@@ -79,7 +79,7 @@ ADTypes = "0.2, 1"
 Adapt = "3.0, 4"
 ArrayInterface = "7"
 DataStructures = "0.18"
-DiffEqBase = "6.154"
+DiffEqBase = "6.166.1"
 DocStringExtensions = "0.9"
 EnumX = "1"
 ExponentialUtilities = "1"
@@ -133,7 +133,7 @@ PrecompileTools = "1"
 Preferences = "1.3"
 RecursiveArrayTools = "2.36, 3"
 Reexport = "1.0"
-SciMLBase = "2.69"
+SciMLBase = "2.78"
 SciMLOperators = "0.3"
 SciMLStructures = "1"
 SimpleNonlinearSolve = "1, 2"

lib/OrdinaryDiffEqCore/src/misc_utils.jl

@@ -147,6 +147,10 @@ function _bool_to_ADType(::Val{false}, _, ::Val{FD}) where {FD}
 end
 
 # Functions to get ADType type from Bool or ADType object, or ADType type
+function _process_AD_choice(ad_alg::Bool, CS::Int, ::Val{FD}) where {FD}
+    return _bool_to_ADType(Val(ad_alg), Val{CS}(), Val{FD}()), Val{CS}(), Val{FD}()
+end
+
 function _process_AD_choice(ad_alg::Bool, ::Val{CS}, ::Val{FD}) where {CS, FD}
     return _bool_to_ADType(Val(ad_alg), Val{CS}(), Val{FD}()), Val{CS}(), Val{FD}()
 end
@@ -162,6 +166,17 @@ function _process_AD_choice(
     return ad_alg, Val{_CS}(), Val{FD}()
 end
 
+function _process_AD_choice(
+        ad_alg::AutoForwardDiff{CS}, CS2::Int, ::Val{FD}) where {CS, FD}
+    # Non-default `chunk_size`
+    if CS2 != 0
+        @warn "The `chunk_size` keyword is deprecated. Please use an `ADType` specifier. For now defaulting to using `AutoForwardDiff` with `chunksize=$(CS2)`."
+        return _bool_to_ADType(Val{true}(), Val{CS2}(), Val{FD}()), Val{CS2}(), Val{FD}()
+    end
+    _CS = CS === nothing ? 0 : CS
+    return ad_alg, Val{_CS}(), Val{FD}()
+end
+
 function _process_AD_choice(
         ad_alg::AutoFiniteDiff{FD}, ::Val{CS}, ::Val{FD2}) where {FD, CS, FD2}
     # Non-default `diff_type`

lib/OrdinaryDiffEqRosenbrock/src/rosenbrock_caches.jl

@@ -763,8 +763,9 @@ function alg_cache(
     du1 = zero(rate_prototype)
     du2 = zero(rate_prototype)
 
-    dtC = similar(tab.C)
-    dtd = similar(tab.d)
+    # Promote t-type for AD
+    dtC = zero(tab.C) .* dt
+    dtd = zero(tab.d) .* dt
 
     # Initialize other variables
     fsalfirst = zero(rate_prototype)