diff --git a/tests/sparse_bcoo_bcsr_test.py b/tests/sparse_bcoo_bcsr_test.py index 12088db7f..e839bacbe 100644 --- a/tests/sparse_bcoo_bcsr_test.py +++ b/tests/sparse_bcoo_bcsr_test.py @@ -1514,8 +1514,8 @@ class BCOOTest(sptu.SparseTestCase): args_maker_sp_de = lambda: [sprng(lhs_shape, lhs_dtype, n_batch=n_batch_lhs), jnp.array(rng(rhs_shape, rhs_dtype))] - tol = {np.float64: 1E-7, np.complex128: 1E-7, - np.float32: 1E-6, np.complex64: 1E-6} + tol = {np.float64: 1E-4, np.complex128: 1E-7, + np.float32: 1E-4, np.complex64: 1E-6} with jtu.strict_promotion_if_dtypes_match([lhs_dtype, rhs_dtype]): self._CheckAgainstDense(operator.matmul, operator.matmul, args_maker_de_sp, tol=tol)