Skip to content

Commit bd380f7

Browse files
fix
1 parent d2abe2e commit bd380f7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/cuda/layers/msgpass.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ end
1515
NewCudaLayer(m, n) = NewCudaLayer(randn(T, m,n))
1616
@functor NewCudaLayer
1717

18-
(l::NewCudaLayer)(X) = propagate(l, X, +)
18+
(l::NewCudaLayer)(X) = GraphNeuralNetworks.propagate(l, X, +)
1919
GraphNeuralNetworks.message(n::NewCudaLayer, x_i, x_j, e_ij) = n.weight * x_j
2020
GraphNeuralNetworks.update(::NewCudaLayer, m, x) = m
2121

0 commit comments

Comments
 (0)