From 2eddfc8bfd1cf4cd489b2c69f4cb44e1d4fec459 Mon Sep 17 00:00:00 2001 From: CarloLucibello Date: Sat, 30 Nov 2024 19:32:42 +0100 Subject: [PATCH] chebconv --- GraphNeuralNetworks/test/layers/conv.jl | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/GraphNeuralNetworks/test/layers/conv.jl b/GraphNeuralNetworks/test/layers/conv.jl index 2167cf47a..8ea2a48db 100644 --- 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