[sparse] adjust tolerance on bcoo_dot_general_sampled

PiperOrigin-RevId: 513342523
google · Mar 1, 2023 · abc6c9b · abc6c9b
1 parent 3abae68
commit abc6c9b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,