Skip to content

Commit 33201f9

Browse files
fix parallel
1 parent 1639349 commit 33201f9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/layers/basic.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ applylayer(l, g::GNNGraph, x) = l(x)
6969
applylayer(l::GNNLayer, g::GNNGraph, x) = l(g, x)
7070

7171
# Handle Flux.Parallel
72+
applylayer(l::Parallel, g::GNNGraph) = GNNGraph(g, ndata=applylayer(l, g, node_features(g)))
7273
applylayer(l::Parallel, g::GNNGraph, x::AbstractArray) = mapreduce(f -> applylayer(f, g, x), l.connection, l.layers)
7374
applylayer(l::Parallel, g::GNNGraph, xs::Vararg{<:AbstractArray}) = mapreduce((f, x) -> applylayer(f, g, x), l.connection, l.layers, xs)
7475
applylayer(l::Parallel, g::GNNGraph, xs::Tuple) = applylayer(l, g, xs...)

0 commit comments

Comments
 (0)