diff --git a/src/norecompile.jl b/src/norecompile.jl index 1f0406050..fa6992cc8 100644 --- a/src/norecompile.jl +++ b/src/norecompile.jl @@ -64,7 +64,7 @@ function wrapfun_iip(ff, Tuple{dualT1, T2, T3, dualT4}, # tgrad ) - iip_returnlists = ntuple(x -> Nothing, 4) + iip_returnlists = ntuple(x -> Nothing, length(iip_arglists) fwt = map(iip_arglists, iip_returnlists) do A, R FunctionWrappersWrappers.FunctionWrappers.FunctionWrapper{R, A}(Void(ff))