Skip to content

Commit 3688442

Browse files
fix test
1 parent bab43ee commit 3688442

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/layers/conv.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,9 @@
4848
@test_broken test_layer(l, g, rtol=1e-5, broken_grad_fields=[:weight], test_gpu=false)
4949
else
5050
test_layer(l, g, rtol=1e-5, broken_grad_fields=[:weight], test_gpu=false)
51-
@test_broken test_layer(l, g, rtol=1e-5, broken_grad_fields=[:weight], test_gpu=true)
51+
if TEST_GPU
52+
@test_broken test_layer(l, g, rtol=1e-5, broken_grad_fields=[:weight], test_gpu=true)
53+
end
5254
end
5355
end
5456

0 commit comments

Comments
 (0)