From 4b7892eefb2933d94b676983dfad5f6b55f56987 Mon Sep 17 00:00:00 2001 From: CarloLucibello Date: Sat, 30 Nov 2024 12:38:35 +0100 Subject: [PATCH] broken --- GNNlib/test/msgpass.jl | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/GNNlib/test/msgpass.jl b/GNNlib/test/msgpass.jl index 846765704..5741cdb5a 100644 --- 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