Ring distributed backend (#1784)

2025-10-19 08:10:15 +08:00 · 2025-01-27 22:15:01 -08:00
parent 2235dee906
commit ccb61d7aae
17 changed files with 1078 additions and 44 deletions
--- a/python/tests/mpi_test_distributed.py
+++ b/python/tests/mpi_test_distributed.py
@@ -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: