Skip to content

Commit b96376c

Browse files
CI test (#585)
1 parent 168de8a commit b96376c

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

GNNGraphs/src/query.jl

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -272,13 +272,11 @@ function CRC.rrule(::typeof(adjacency_matrix), g::G, T::DataType;
272272
function adjacency_matrix_pullback_weighted(Δ)
273273
dy = CRC.unthunk(Δ)
274274
s, t = edge_index(g)
275-
@show dy s t
276275
#TODO using CRC.@thunk gives an error
277276
#TODO use gather when https://github.com/FluxML/NNlib.jl/issues/625 is fixed
278-
dw = zeros_like(w)
277+
dw = zeros_like(w, eltype(dy))
279278
idx = CartesianIndex.(s, t) #TODO remove when https://github.com/FluxML/NNlib.jl/issues/626 is fixed
280279
NNlib.gather!(dw, dy, idx)
281-
@show dw
282280
dg = CRC.Tangent{G}(; graph = (CRC.NoTangent(), CRC.NoTangent(), dw))
283281
return (CRC.NoTangent(), dg, CRC.NoTangent())
284282
end

0 commit comments

Comments
 (0)