We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 366bdb8 commit e083ffcCopy full SHA for e083ffc
GNNlib/ext/GNNlibCUDAExt.jl
@@ -3,7 +3,7 @@ module GNNlibCUDAExt
3
using CUDA
4
using Random, Statistics, LinearAlgebra
5
using GNNlib: GNNlib, propagate, copy_xj, e_mul_xj, w_mul_xj
6
-using GNNGraphs: GNNGraph, COO_T, SPARSE_T, to_dense, to_sparse
+using GNNGraphs: GNNGraph, COO_T, SPARSE_T, to_dense, to_sparse, adjacency_matrix
7
using ChainRulesCore: @non_differentiable
8
9
const CUDA_COO_T = Tuple{T, T, V} where {T <: AnyCuArray{<:Integer}, V <: Union{Nothing, AnyCuArray}}
0 commit comments