From abc6c9bf4924f88faf86ddf8c29cb5048aa03c12 Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Wed, 1 Mar 2023 14:28:43 -0800 Subject: [PATCH] [sparse] adjust tolerance on bcoo_dot_general_sampled PiperOrigin-RevId: 513342523 --- tests/sparse_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/sparse_test.py b/tests/sparse_test.py index 01323563d70d..a1da8d097625 100644 --- a/tests/sparse_test.py +++ b/tests/sparse_test.py @@ -1310,8 +1310,8 @@ def f1(x, y, indices): def f2(x, y, indices): return sparse.bcoo_dot_general_sampled(x, y, indices, dimension_numbers=dimension_numbers) - self._CheckAgainstNumpy(f1, f2, args_maker) - self._CompileAndCheck(f2, args_maker) + self._CheckAgainstNumpy(f1, f2, args_maker, tol=MATMUL_TOL) + self._CompileAndCheck(f2, args_maker, tol=MATMUL_TOL) @jtu.sample_product( [dict(n_batch=n_batch, n_dense=n_dense, lhs_shape=lhs_shape,