Skip to content

Commit 876bc5f

Browse files
committed
comment and remove 4th method
1 parent f8cd198 commit 876bc5f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/norecompile.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -59,10 +59,10 @@ function wrapfun_iip(ff,
5959
dualT2 = ArrayInterface.promote_eltype(T2, dualT)
6060
dualT4 = dualgen(T4)
6161

62-
iip_arglists = (Tuple{T1, T2, T3, T4},
63-
Tuple{dualT1, dualT2, T3, T4},
64-
Tuple{dualT1, T2, T3, dualT4},
65-
Tuple{dualT1, dualT2, T3, dualT4})
62+
iip_arglists = (Tuple{T1, T2, T3, T4}, # primal
63+
Tuple{dualT1, dualT2, T3, T4}, # vjp
64+
Tuple{dualT1, T2, T3, dualT4}, # tgrad
65+
)
6666

6767
iip_returnlists = ntuple(x -> Nothing, 4)
6868

0 commit comments

Comments
 (0)