Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 90599e1

Browse files
haojin2eric-haibin-lin
authored andcommitted
further bump up tolerance for sparse dot (#12527)
1 parent 3d83c89 commit 90599e1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/python/unittest/test_sparse_operator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1307,7 +1307,7 @@ def test_infer_forward_stype(lhs_shape, rhs_shape, lhs_density, rhs_density, tra
13071307
rhs = rhs_nd.tostype(rhs_stype)
13081308
out = mx.nd.dot(lhs, rhs, forward_stype=forward_stype,
13091309
transpose_a=trans_a, transpose_b=trans_b)
1310-
assert_almost_equal(out.tostype('default').asnumpy(), out_np, rtol=1e-3, atol=1e-5)
1310+
assert_almost_equal(out.tostype('default').asnumpy(), out_np, rtol=1e-3, atol=1e-4)
13111311
lhs_var = mx.symbol.Variable('lhs', stype=lhs_stype)
13121312
rhs_var = mx.symbol.Variable('rhs', stype=rhs_stype)
13131313
out = mx.symbol.sparse.dot(lhs_var, rhs_var,

0 commit comments

Comments
 (0)