From 8136d49843d19e9e20843e7307292f8f6b0e3ab9 Mon Sep 17 00:00:00 2001 From: Caenorst Date: Wed, 2 Oct 2019 17:47:59 -0400 Subject: [PATCH] fix atol for test_preloaded_multi_sgd (#16356) --- tests/python/gpu/test_operator_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/gpu/test_operator_gpu.py b/tests/python/gpu/test_operator_gpu.py index 7aac23acd549..0f1cd93755c3 100644 --- a/tests/python/gpu/test_operator_gpu.py +++ b/tests/python/gpu/test_operator_gpu.py @@ -396,7 +396,7 @@ def _assert_all_almost_equal(lhs_list, rhs_list, rtol, atol): assert_almost_equal(lhs.asnumpy(), rhs.asnumpy(), rtol=rtol, atol=atol) if dtype == 'float16': rtol = 1e-3 - atol = 1e-3 + atol = 1e-2 else: rtol = 1e-5 atol = 1e-6