fix ambiguity by splitting union

SciML · Dec 31, 2023 · 11ec603 · 11ec603
1 parent cc1960c
commit 11ec603
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/ext/PreallocationToolsSymbolicsExt.jl b/ext/PreallocationToolsSymbolicsExt.jl
@@ -4,21 +4,42 @@ using PreallocationTools
 import PreallocationTools: _restructure, get_tmp
 using Symbolics, ForwardDiff
 
-function get_tmp(dc::Union{FixedSizeDiffCache,DiffCache}, u::Type{X}) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
+function get_tmp(dc::DiffCache, u::Type{X}) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
     if length(dc.du) > length(dc.any_du)
         resize!(dc.any_du, length(dc.du))
     end
     _restructure(dc.du, dc.any_du)
 end
 
-function get_tmp(dc::Union{FixedSizeDiffCache,DiffCache}, u::X) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
+function get_tmp(dc::DiffCache, u::X) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
     if length(dc.du) > length(dc.any_du)
         resize!(dc.any_du, length(dc.du))
     end
     _restructure(dc.du, dc.any_du)
 end
 
-function get_tmp(dc::Union{FixedSizeDiffCache,DiffCache}, u::AbstractArray{X}) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
+function get_tmp(dc::DiffCache, u::AbstractArray{X}) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
+    if length(dc.du) > length(dc.any_du)
+        resize!(dc.any_du, length(dc.du))
+    end
+    _restructure(dc.du, dc.any_du)
+end
+
+function get_tmp(dc::FixedSizeDiffCache, u::Type{X}) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
+    if length(dc.du) > length(dc.any_du)
+        resize!(dc.any_du, length(dc.du))
+    end
+    _restructure(dc.du, dc.any_du)
+end
+
+function get_tmp(dc::FixedSizeDiffCache, u::X) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
+    if length(dc.du) > length(dc.any_du)
+        resize!(dc.any_du, length(dc.du))
+    end
+    _restructure(dc.du, dc.any_du)
+end
+
+function get_tmp(dc::FixedSizeDiffCache, u::AbstractArray{X}) where {T,N, X<: ForwardDiff.Dual{T, Num, N}}
     if length(dc.du) > length(dc.any_du)
         resize!(dc.any_du, length(dc.du))
     end