Skip to content

Commit 082b929

Browse files
grouped conv + gpu = :(
1 parent 0524be9 commit 082b929

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

test/cuda/layers.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,13 @@ function gpu_gradtest(name::String, layers::Vector, x_cpu = nothing, args...; te
5050
@test isnothing(xg_gpu)
5151
else
5252
if layer === GroupedConvTranspose
53-
@test Array(xg_gpu) xg_cpu rtol=2f-2 atol=1f-3
53+
if VERSION >= v"1.7"
54+
@test_broken Array(xg_gpu) xg_cpu rtol = 2f-2 atol = 1f-3
55+
else
56+
@test Array(xg_gpu) xg_cpu rtol = 2f-2 atol = 1f-3
57+
end
5458
else
55-
@test Array(xg_gpu) xg_cpu rtol=1f-3 atol=1f-3
59+
@test Array(xg_gpu) xg_cpu rtol = 1f-3 atol = 1f-3
5660
end
5761
end
5862
end

0 commit comments

Comments
 (0)