Skip to content

Commit ad2350e

Browse files
Apply suggestion from @CarloLucibello
1 parent 84d49d1 commit ad2350e

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

GraphNeuralNetworks/test/test_module.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ function test_gradients(
125125
if test_mooncake
126126
# Mooncake gradient with respect to input, compared against Zygote.
127127
loss_mc_x = (xs...) -> loss(f, graph, xs...)
128+
# TODO error without `invokelatest` when using TestItemRunner
128129
_cache_x = Base.invokelatest(Mooncake.prepare_gradient_cache, loss_mc_x, xs...)
129130
y_mc, g_mc = Base.invokelatest(Mooncake.value_and_gradient!!, _cache_x, loss_mc_x, xs...)
130131
@assert isapprox(y, y_mc; rtol, atol)

0 commit comments

Comments
 (0)