get_graph_type

JuliaGraphs · Nov 30, 2024 · d207de4 · d207de4
1 parent 55fe50b
commit d207de4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/GNNlib/test/msgpass.jl b/GNNlib/test/msgpass.jl
@@ -195,9 +195,11 @@ end
                 g = set_edge_weight(g, w)
                 return propagate(w_mul_xj, g, +, xj = x)
             end
+            dev = gpu_device(force=true)
+            broken = get_graph_type(g) == :sparse && dev isa AMDGPUDevice
             @test test_gradients(
                 f, g, g.x, w; test_gpu=true, test_grad_f=false, compare_finite_diff=false
-            ) broken=true
+            ) broken=broken
         end
     end
 end