broken

JuliaGraphs · Nov 30, 2024 · 4b7892e · 4b7892e
1 parent 503b800
commit 4b7892e
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/GNNlib/test/msgpass.jl b/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