Skip to content

Commit e083ffc

Browse files
fix nnlibcudaext
1 parent 366bdb8 commit e083ffc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

GNNlib/ext/GNNlibCUDAExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ module GNNlibCUDAExt
33
using CUDA
44
using Random, Statistics, LinearAlgebra
55
using GNNlib: GNNlib, propagate, copy_xj, e_mul_xj, w_mul_xj
6-
using GNNGraphs: GNNGraph, COO_T, SPARSE_T, to_dense, to_sparse
6+
using GNNGraphs: GNNGraph, COO_T, SPARSE_T, to_dense, to_sparse, adjacency_matrix
77
using ChainRulesCore: @non_differentiable
88

99
const CUDA_COO_T = Tuple{T, T, V} where {T <: AnyCuArray{<:Integer}, V <: Union{Nothing, AnyCuArray}}

0 commit comments

Comments
 (0)