zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-02 01:08:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve the ring backend initialization

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-07-11 15:31:28 -07:00
parent 8347575ba1
commit a9c720e8cd
3 changed files with 38 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/tests/mpi_test_distributed.py
View File

@@ -4,6 +4,7 @@ import unittest
import mlx.core as mx
import mlx_distributed_tests
import mlx_tests
class TestMPIDistributed(mlx_distributed_tests.MLXDistributedCommonTestCase):
@@ -150,4 +151,4 @@ class TestMPIDistributed(mlx_distributed_tests.MLXDistributedCommonTestCase):
if __name__ == "__main__":
unittest.main()
mlx_tests.MLXTestRunner()

1
python/tests/ring_test_distributed.py
View File

@@ -4,6 +4,7 @@ import unittest
import mlx.core as mx
import mlx_distributed_tests
import mlx_tests
class TestRingDistributed(mlx_distributed_tests.MLXDistributedCommonTestCase):