further bump up tolerance for sparse dot (apache#12527)

anirudh2290 · Sep 19, 2018 · 5d3ec49 · 5d3ec49
1 parent f66c72f
commit 5d3ec49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,