fixup! fix: consider type of t0 in promote_u0

SciML · May 27, 2024 · 986a43d · 986a43d
1 parent 9c375c7
commit 986a43d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/forwarddiff.jl b/src/forwarddiff.jl
@@ -266,6 +266,9 @@ end
 
 @inline function promote_u0(u0, p, t0)
     Tu = eltype(u0)
+    if Tu <: ForwardDiff.Dual
+        return u0
+    end
     Tp = anyeltypedual(p)
     if Tp == Any
         Tp = Tu
@@ -284,6 +287,9 @@ end
 
 @inline function promote_u0(u0::AbstractArray{<:Complex}, p, t0)
     Tu = real(eltype(u0))
+    if Tu <: ForwardDiff.Dual
+        return u0
+    end
     Tp = anyeltypedual(p)
     if Tp == Any
         Tp = Tu
@@ -293,7 +299,7 @@ end
         Tt = Tu
     end
     Tcommon = promote_type(eltype(u0), Tp, Tt)
-    return if Tcommon <: ForwardDiff.Dual
+    return if real(Tcommon) <: ForwardDiff.Dual
         Tcommon.(u0)
     else
         u0