Skip to content

Commit 87a82b5

Browse files
authored
Try to revert GPU tests
1 parent edaa9aa commit 87a82b5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/gpu/simple_gpu.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,14 +95,14 @@ Random.seed!(100)
9595

9696
@testset "quadreg" begin
9797
# use a different reg parameter
98-
ε_quad = 0.5f0
98+
ε_quad = 1.0f0
9999
γ = quadreg(
100100
cu_μ, cu_ν, cu_C, ε_quad, QuadraticOTNewton(0.1f0, 0.5f0, 1.0f-5, 50)
101101
)
102102
# compare with results on the CPU
103103
@test convert(Array, γ)
104104
quadreg(μ, ν, C, ε_quad, QuadraticOTNewton(0.1f0, 0.5f0, 1.0f-5, 50)) atol =
105-
1.0f-3 rtol = 1.0f-3
105+
1.0f-4 rtol = 1.0f-4
106106
end
107107
end
108108
end

0 commit comments

Comments
 (0)