Fix RandomBits::is_equivalent to include width (#2978)

Co-authored-by: KD2YCU <[email protected]> Co-authored-by: Angelos Katharopoulos <[email protected]> Co-authored-by: Awni Hannun <[email protected]>
2026-01-13 12:42:37 -08:00
parent a8197795f5
commit 4160ec10f7
3 changed files with 15 additions and 1 deletions
@@ -297,6 +297,7 @@ class TestBlas(mlx_tests.MLXTestCase):
            self.assertTrue(np.allclose(out_mlx, out_npy, atol=1e-5))

    def test_matrix_vector(self):
+        mx.random.seed(0)
        for dtype in self.dtypes:
            with self.subTest(dtype=dtype):
                np_dtype = getattr(np, dtype)