Skip to content

Commit 16a55b3

Browse files
Merge pull request #3332 from ChrisRackauckas-Claude/fix-fww-v1-compat
Fix FunctionWrappersWrappers v1.0 compat in DiffEqBase ForwardDiff ext
2 parents b74f047 + 5c80e4c commit 16a55b3

1 file changed

Lines changed: 3 additions & 12 deletions

File tree

lib/DiffEqBase/ext/DiffEqBaseForwardDiffExt.jl

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,7 @@ function wrapfun_iip(
8080

8181
iip_returnlists = ntuple(x -> Nothing, 4)
8282

83-
fwt = map(iip_arglists, iip_returnlists) do A, R
84-
FunctionWrappersWrappers.FunctionWrappers.FunctionWrapper{R, A}(Void(ff))
85-
end
86-
return FunctionWrappersWrappers.FunctionWrappersWrapper{typeof(fwt), false}(fwt)
83+
return FunctionWrappersWrappers.FunctionWrappersWrapper(Void(ff), iip_arglists, iip_returnlists)
8784
end
8885

8986
# 3-arg version: compile FunctionWrapper variants with the specified chunk size.
@@ -115,10 +112,7 @@ function wrapfun_iip(
115112

116113
iip_returnlists = ntuple(x -> Nothing, 4)
117114

118-
fwt = map(iip_arglists, iip_returnlists) do A, R
119-
FunctionWrappersWrappers.FunctionWrappers.FunctionWrapper{R, A}(Void(ff))
120-
end
121-
return FunctionWrappersWrappers.FunctionWrappersWrapper{typeof(fwt), false}(fwt)
115+
return FunctionWrappersWrappers.FunctionWrappersWrapper(Void(ff), iip_arglists, iip_returnlists)
122116
end
123117

124118
const iip_arglists_default = (
@@ -146,10 +140,7 @@ const iip_arglists_default = (
146140
const iip_returnlists_default = ntuple(x -> Nothing, length(iip_arglists_default))
147141

148142
function wrapfun_iip(@nospecialize(ff))
149-
fwt = map(iip_arglists_default, iip_returnlists_default) do A, R
150-
FunctionWrappersWrappers.FunctionWrappers.FunctionWrapper{R, A}(Void(ff))
151-
end
152-
return FunctionWrappersWrappers.FunctionWrappersWrapper{typeof(fwt), false}(fwt)
143+
return FunctionWrappersWrappers.FunctionWrappersWrapper(Void(ff), iip_arglists_default, iip_returnlists_default)
153144
end
154145

155146
function promote_tspan(u0::AbstractArray{<:ForwardDiff.Dual}, p, tspan, prob, kwargs)

0 commit comments

Comments
 (0)