Skip to content

Commit 205ec52

Browse files
committedFeb 25, 2021
fix
1 parent 0be4b59 commit 205ec52

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed
 

‎test/test_utils.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ gradtest(f, dims::IntOrTuple...; kw...) =
88
Compare numerical gradient and automatic gradient
99
given by Zygote. `f` has to be a scalar valued function.
1010
11-
Applies also `ChainRulesTestUtils.rrule_test` if the rrule for `f` is explicitly defined.
11+
Applies also `ChainRulesTestUtils.test_rrule` if the rrule for `f` is explicitly defined.
1212
"""
1313
function gradtest(f, xs...; atol=1e-6, rtol=1e-6, fkwargs=NamedTuple(),
1414
check_rrule=false,
@@ -20,7 +20,7 @@ function gradtest(f, xs...; atol=1e-6, rtol=1e-6, fkwargs=NamedTuple(),
2020
simil(x) = x isa Number ? randn(rng, typeof(x)) : randn!(rng, similar(x))
2121
= simil(y)
2222
xx̄s = [x simil(x) for x in xs]
23-
rrule_test(f, xx̄s...; fkwargs=fkwargs, output_tangent=ȳ)
23+
test_rrule(f, xx̄s...; fkwargs=fkwargs, output_tangent=ȳ)
2424
end
2525

2626
if check_broadcast

0 commit comments

Comments
 (0)