Merge pull request #107 from yuehhua/develop

Bump CUDA to v2.0

Author: yuehhua

Project.toml

@@ -25,7 +25,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
 
 [compat]
-CUDA = "1.2"
+CUDA = "1.2 - 2.0"
 DataStructures = "0.17 - 0.18"
 FillArrays = "0.8 - 0.9"
 Flux = "0.10 - 0.11"

src/cuda/msgpass.jl

@@ -1,5 +1,5 @@
 @inline function update_batch_edge(mp::T, adj, E::AbstractMatrix, X::CuMatrix) where {T<:MessagePassing}
-    E = convert(typeof(X), E)
+    E = convert(typeof(X), Matrix(E))
     update_batch_edge(mp, adj, E, X)
 end
 

test/layers/msgpass.jl

@@ -25,7 +25,7 @@ l = NewLayer(out_channel, in_channel)
     @testset "no message or update" begin
         fg_ = l(fg)
 
-        @test graph(fg_) === adj
+        @test graph(fg_) == adj
         @test size(node_feature(fg_)) == (in_channel, num_V)
         @test size(edge_feature(fg_)) == (in_channel, 2*num_E)
         @test size(global_feature(fg_)) == (0,)
@@ -35,7 +35,7 @@ l = NewLayer(out_channel, in_channel)
     @testset "message function" begin
         fg_ = l(fg)
 
-        @test graph(fg_) === adj
+        @test graph(fg_) == adj
         @test size(node_feature(fg_)) == (out_channel, num_V)
         @test size(edge_feature(fg_)) == (out_channel, 2*num_E)
         @test size(global_feature(fg_)) == (0,)
@@ -45,7 +45,7 @@ l = NewLayer(out_channel, in_channel)
     @testset "message and update" begin
         fg_ = l(fg)
 
-        @test graph(fg_) === adj
+        @test graph(fg_) == adj
         @test size(node_feature(fg_)) == (out_channel, num_V)
         @test size(edge_feature(fg_)) == (out_channel, 2*num_E)
         @test size(global_feature(fg_)) == (0,)