From 876bc5f3813bd6d3155622b3fb420fe652ccc1c4 Mon Sep 17 00:00:00 2001 From: oscarddssmith Date: Mon, 6 Nov 2023 15:33:47 -0500 Subject: [PATCH] comment and remove 4th method --- src/norecompile.jl | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/norecompile.jl b/src/norecompile.jl index ecee30d44..1f0406050 100644 --- a/src/norecompile.jl +++ b/src/norecompile.jl @@ -59,10 +59,10 @@ function wrapfun_iip(ff, dualT2 = ArrayInterface.promote_eltype(T2, dualT) dualT4 = dualgen(T4) - iip_arglists = (Tuple{T1, T2, T3, T4}, - Tuple{dualT1, dualT2, T3, T4}, - Tuple{dualT1, T2, T3, dualT4}, - Tuple{dualT1, dualT2, T3, dualT4}) + iip_arglists = (Tuple{T1, T2, T3, T4}, # primal + Tuple{dualT1, dualT2, T3, T4}, # vjp + Tuple{dualT1, T2, T3, dualT4}, # tgrad + ) iip_returnlists = ntuple(x -> Nothing, 4)