broken

Author: CarloLucibello

GNNlib/test/msgpass.jl

@@ -171,7 +171,9 @@ end
             dev = gpu_device(force=true)
             broken = get_graph_type(g) == :sparse && dev isa AMDGPUDevice
             f(g, x) = propagate(copy_xj, g, +, xj = x)
-            test_gradients(f, g, g.x; test_gpu=true, test_grad_f=false, compare_finite_diff=false)
+            @test test_gradients(
+                f, g, g.x; test_gpu=true, test_grad_f=false, compare_finite_diff=false
+            ) broken=broken
         end
     end
 
@@ -180,7 +182,9 @@ end
             dev = gpu_device(force=true)
             broken = get_graph_type(g) == :sparse && dev isa AMDGPUDevice
             f(g, x) = propagate(copy_xj, g, mean, xj = x)
-            test_gradients(f, g, g.x; test_gpu=true, test_grad_f=false, compare_finite_diff=false)
+            @test test_gradients(
+                f, g, g.x; test_gpu=true, test_grad_f=false, compare_finite_diff=false
+            ) broken=broken
         end
     end
 
@@ -190,7 +194,9 @@ end
             broken = get_graph_type(g) == :sparse && dev isa AMDGPUDevice
             e = rand(Float32, size(g.x, 1), g.num_edges)
             f(g, x, e) = propagate(e_mul_xj, g, +; xj = x, e)
-            test_gradients(f, g, g.x, e; test_gpu=true, test_grad_f=false, compare_finite_diff=false)
+            @test test_gradients(
+                f, g, g.x, e; test_gpu=true, test_grad_f=false, compare_finite_diff=false
+            ) broken=broken
         end
     end