@@ -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)
8784end
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)
122116end
123117
124118const iip_arglists_default = (
@@ -146,10 +140,7 @@ const iip_arglists_default = (
146140const iip_returnlists_default = ntuple (x -> Nothing, length (iip_arglists_default))
147141
148142function 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)
153144end
154145
155146function promote_tspan (u0:: AbstractArray{<:ForwardDiff.Dual} , p, tspan, prob, kwargs)
0 commit comments