From 90599e1038a4ff6604e9ed0d55dc274c2df635f8 Mon Sep 17 00:00:00 2001 From: Hao Jin Date: Tue, 11 Sep 2018 20:35:45 -0700 Subject: [PATCH] further bump up tolerance for sparse dot (#12527) --- tests/python/unittest/test_sparse_operator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_sparse_operator.py b/tests/python/unittest/test_sparse_operator.py index b4783bd23820..57808248b081 100644 --- a/tests/python/unittest/test_sparse_operator.py +++ b/tests/python/unittest/test_sparse_operator.py @@ -1307,7 +1307,7 @@ def test_infer_forward_stype(lhs_shape, rhs_shape, lhs_density, rhs_density, tra rhs = rhs_nd.tostype(rhs_stype) out = mx.nd.dot(lhs, rhs, forward_stype=forward_stype, transpose_a=trans_a, transpose_b=trans_b) - assert_almost_equal(out.tostype('default').asnumpy(), out_np, rtol=1e-3, atol=1e-5) + assert_almost_equal(out.tostype('default').asnumpy(), out_np, rtol=1e-3, atol=1e-4) lhs_var = mx.symbol.Variable('lhs', stype=lhs_stype) rhs_var = mx.symbol.Variable('rhs', stype=rhs_stype) out = mx.symbol.sparse.dot(lhs_var, rhs_var,