Skip to content

Commit d4c95b1

Browse files
author
KDr2
committed
test case for tape length (tracking or not)
1 parent 9259b73 commit d4c95b1

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

test/ChainRulesTests.jl

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,18 @@ ReverseDiff.@grad_from_chainrules g(x::ReverseDiff.TrackedArray, y::ReverseDiff.
9393

9494
end
9595

96+
### Tape test
97+
@testset "Tape test: Ensure ordinary call is not tracked" begin
98+
tp = ReverseDiff.InstructionTape()
99+
100+
f(x) = sum(2x .+ g([1, 2], [3, 4]))
101+
x = rand(3, 3)
102+
xt = ReverseDiff.track(copy(x), tp)
103+
# record
104+
yt = f(xt)
105+
@test length(tp) == 3 # sum, broadcast+, broadcast*, but not `g`
106+
end
107+
96108
### Functions with varargs and kwargs
97109
# Varargs
98110
f_vararg(x, args...) = sum(4x .+ sum(args))

0 commit comments

Comments
 (0)