Skip to content

Commit ac3cdfa

Browse files
committed
adjust tolerances of gpu tests
1 parent bc488c7 commit ac3cdfa

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

test/cuda/layers.jl

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,11 @@ function gpu_gradtest(name::String, layers::Vector, x_cpu = nothing, args...; te
5454
if isnothing(xg_cpu)
5555
@test isnothing(xg_gpu)
5656
else
57-
@test Array(xg_gpu) xg_cpu rtol=2f-2 atol=1f-3
57+
if layer === GroupedConvTranspose
58+
@test Array(xg_gpu) xg_cpu rtol=2f-2 atol=1f-3
59+
else
60+
@test Array(xg_gpu) xg_cpu rtol=1f-3 atol=1f-3
61+
end
5862
end
5963
end
6064
@test gs_gpu isa Flux.Zygote.Grads

0 commit comments

Comments
 (0)