Skip to content

Commit 0f45431

Browse files
fix
1 parent 0920e2c commit 0f45431

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

GNNGraphs/test/test_utils.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,12 @@
11
using ChainRulesTestUtils, FiniteDifferences, Zygote, Adapt, CUDA
22
CUDA.allowscalar(false)
33

4+
# Using this until https://github.com/JuliaDiff/FiniteDifferences.jl/issues/188 is fixed
5+
function FiniteDifferences.to_vec(x::Integer)
6+
Integer_from_vec(v) = x
7+
return Int[x], Integer_from_vec
8+
end
9+
410
function ngradient(f, x...)
511
fdm = central_fdm(5, 1)
612
return FiniteDifferences.grad(fdm, f, x...)

0 commit comments

Comments
 (0)