Tune float32 tests

Author: jessegrabowski

tests/tensor/test_slinalg.py

@@ -417,7 +417,12 @@ def test_correctness(self, b_shape: tuple[int], lower, trans, unit_diagonal):
             unit_diagonal=unit_diagonal,
         )
 
-        np.testing.assert_allclose(x_pt, x_sp)
+        np.testing.assert_allclose(
+            x_pt,
+            x_sp,
+            atol=1e-8 if config.floatX == "float64" else 1e-4,
+            rtol=1e-8 if config.floatX == "float64" else 1e-4,
+        )
 
     @pytest.mark.parametrize(
         "b_shape", [(5, 1), (5,), (5, 5)], ids=["b_col_vec", "b_vec", "b_matrix"]
@@ -426,6 +431,9 @@ def test_correctness(self, b_shape: tuple[int], lower, trans, unit_diagonal):
     @pytest.mark.parametrize("trans", [0, 1])
     @pytest.mark.parametrize("unit_diagonal", [True, False])
     def test_solve_triangular_grad(self, b_shape, lower, trans, unit_diagonal):
+        if config.floatX == "float32":
+            pytest.skip(reason="Not enough precision in float32 to get a good gradient")
+
         rng = np.random.default_rng(utt.fetch_seed())
         A_val = rng.normal(size=(5, 5)).astype(config.floatX)
         b_val = rng.normal(size=b_shape).astype(config.floatX)