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

Co-authored-by: KD2YCU <me@kd2ycu.com>
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
MillaFleurs
2026-01-13 15:42:37 -05:00
committed by GitHub
parent a8197795f5
commit 4160ec10f7
3 changed files with 15 additions and 1 deletions
+1
View File
@@ -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)