Skip to content

Commit 8a1283e

Browse files
author
Katharine Hyatt
committed
Fix another issue
1 parent 7ee7322 commit 8a1283e

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

test/testsuite/mooncake.jl

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,13 @@ function _get_copying_derivative(f, rrule, A, ΔA, args, Δargs, ::Nothing, rdat
7777
else
7878
copyto!(Mooncake.tangent(copy_out), dargs_copy)
7979
end
80-
@test Mooncake.primal(copy_out) args
80+
if args isa Tuple
81+
for (arg, out) in zip(args, Mooncake.primal(copy_out))
82+
@test out arg
83+
end
84+
else
85+
@test Mooncake.primal(copy_out) args
86+
end
8187
copy_pb!!(rdata)
8288
return dA_copy, Mooncake.tangent(copy_out)
8389
end

0 commit comments

Comments
 (0)