chebconv

JuliaGraphs · Nov 30, 2024 · 2eddfc8 · 2eddfc8
1 parent efefbe0
commit 2eddfc8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/GraphNeuralNetworks/test/layers/conv.jl b/GraphNeuralNetworks/test/layers/conv.jl
@@ -102,9 +102,11 @@ end
     l = ChebConv(D_IN => D_OUT, k)
     for g in TEST_GRAPHS
         has_isolated_nodes(g) && continue
-        g.graph isa AbstractSparseMatrix && continue
-        @test size(l(g, g.x)) == (D_OUT, g.num_nodes)
-        test_gradients(l, g, g.x, rtol = RTOL_LOW, test_gpu = true, compare_finite_diff = false)
+        broken = get_graph_type(g) == :sparse || gpu_device() isa AMDGPUDevice
+        @test size(l(g, g.x)) == (D_OUT, g.num_nodes) broken=broken
+        @test test_gradients(
+            l, g, g.x, rtol = RTOL_LOW, test_gpu = true, compare_finite_diff = false
+        ) broken=broken
     end   
 end