Skip to content

Commit 0e6f3d5

Browse files
add check for GroupedConvTranspose
1 parent 082b929 commit 0e6f3d5

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

test/cuda/layers.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -45,16 +45,16 @@ function gpu_gradtest(name::String, layers::Vector, x_cpu = nothing, args...; te
4545

4646
# test
4747
if test_cpu
48-
@test y_gpu y_cpu rtol=1f-3 atol=1f-3
48+
if VERSION >= v"1.7" && layer === GroupedConvTranspose && args[end] == selu
49+
@test_broken y_gpu y_cpu rtol=1f-3 atol=1f-3
50+
else
51+
@test y_gpu y_cpu rtol=1f-3 atol=1f-3
52+
end
4953
if isnothing(xg_cpu)
5054
@test isnothing(xg_gpu)
5155
else
5256
if layer === GroupedConvTranspose
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
57+
@test Array(xg_gpu) xg_cpu rtol = 2f-2 atol = 1f-3
5858
else
5959
@test Array(xg_gpu) xg_cpu rtol = 1f-3 atol = 1f-3
6060
end

0 commit comments

Comments
 (0)