Cross platform libmpi loading (#1975)

This commit is contained in:
Angelos Katharopoulos
2025-03-21 11:23:10 -07:00
committed by GitHub
parent 4e1994e9d7
commit d343782c8b
5 changed files with 54 additions and 21 deletions

View File

@@ -89,6 +89,7 @@ jobs:
pip install numpy
sudo apt-get update
sudo apt-get install libblas-dev liblapack-dev liblapacke-dev
sudo apt-get install openmpi-bin openmpi-common libopenmpi-dev
- run:
name: Install Python package
command: |
@@ -110,6 +111,8 @@ jobs:
name: Run Python tests
command: |
python3 -m unittest discover python/tests -v
mpirun --bind-to none -host localhost:8 -np 8 python python/tests/mpi_test_distributed.py
mlx.launch --verbose -n 8 python/tests/ring_test_distributed.py
- run:
name: Build CPP only
command: |