Ring distributed backend (#1784)

2025-01-27 22:15:01 -08:00
parent 2235dee906
commit ccb61d7aae
17 changed files with 1078 additions and 44 deletions
@@ -34,6 +34,8 @@ class TestDistributed(mlx_tests.MLXTestCase):
            mx.int32,
            mx.uint32,
            mx.float32,
+            mx.float16,
+            mx.bfloat16,
            mx.complex64,
        ]
        for dt in dtypes: