unbrake chebconv gpu

JuliaGraphs · Aug 5, 2024 · 3ce025b · 3ce025b
1 parent ef22e9a
commit 3ce025b
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/test/layers/conv.jl b/test/layers/conv.jl
@@ -84,12 +84,8 @@ end
     @test l.k == k
     for g in test_graphs
         g = add_self_loops(g)
-        test_layer(l, g, rtol = RTOL_HIGH, test_gpu = false,
+        test_layer(l, g, rtol = RTOL_HIGH, test_gpu = TEST_GPU,
                     outsize = (out_channel, g.num_nodes))
-        if TEST_GPU
-            @test_broken test_layer(l, g, rtol = RTOL_HIGH, test_gpu = true,
-                                    outsize = (out_channel, g.num_nodes))
-        end
     end
 
     @testset "bias=false" begin