Skip to content

Commit 4d9ddac

Browse files
committed
Keep old gather/scatter implementation for COO_T
1 parent f6ff8e3 commit 4d9ddac

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

GNNlib/ext/GNNlibCUDAExt.jl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,14 @@ using GNNGraphs: GNNGraph, COO_T, SPARSE_T
77

88
###### PROPAGATE SPECIALIZATIONS ####################
99

10+
## COPY_XJ
11+
12+
## avoid the fast path on gpu until we have better cuda support
13+
function GNNlib.propagate(::typeof(copy_xj), g::GNNGraph{COO_T}, ::typeof(+),
14+
xi, xj::AnyCuMatrix, e)
15+
propagate((xi, xj, e) -> copy_xj(xi, xj, e), g, +, xi, xj, e)
16+
end
17+
1018
## E_MUL_XJ
1119

1220
## avoid the fast path on gpu until we have better cuda support

0 commit comments

Comments
 (0)