From 3ce025b3395b9c18be7a3303ad81c6291c1247d5 Mon Sep 17 00:00:00 2001 From: CarloLucibello Date: Mon, 5 Aug 2024 15:42:46 +0200 Subject: [PATCH] unbrake chebconv gpu --- test/layers/conv.jl | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/test/layers/conv.jl b/test/layers/conv.jl index 224b98697..4c4827d2e 100644 --- a/test/layers/conv.jl +++ b/test/layers/conv.jl @@ -84,12 +84,8 @@ end @test l.k == k for g in test_graphs g = add_self_loops(g) - test_layer(l, g, rtol = RTOL_HIGH, test_gpu = false, + test_layer(l, g, rtol = RTOL_HIGH, test_gpu = TEST_GPU, outsize = (out_channel, g.num_nodes)) - if TEST_GPU - @test_broken test_layer(l, g, rtol = RTOL_HIGH, test_gpu = true, - outsize = (out_channel, g.num_nodes)) - end end @testset "bias=false" begin