Skip to content

Commit

Permalink
add mpi4py action here
Browse files Browse the repository at this point in the history
Signed-off-by: Howard Pritchard <howardp@lanl.gov>
  • Loading branch information
hppritcha committed Jan 3, 2024
1 parent 9c69e5a commit 9be4d0c
Showing 1 changed file with 36 additions and 0 deletions.
36 changes: 36 additions & 0 deletions .github/workflows/ompi_mpi4py.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
name: MPI4PY Sanity check CI

on: [pull_request]

env:
ROCM_VER: 5-4
PYTHONPATH: $PYTHONPATH:$PWD/mpi4py-3.1.5/
jobs:
run_mpi4py:
runs-on: ubuntu-latest
steps:
- name: Install dependencies
run: |
sudo apt update
sudo apt install -y --no-install-recommends wget lsb-core software-properties-common curl pip3
pip3 install cython
- uses: actions/checkout@v3
with:
submodules: recursive
- name: Build Open MPI
run: |
./autogen.pl
./configure --disable-mpi-fortran
make -j install
- name: Build MPI4py
run: |
wget https://github.com/mpi4py/mpi4py/releases/download/3.1.5/mpi4py-3.1.5.tar.gz
tar -zxf mpi4py-3.1.5.tar.gz
cd mpi4py-3.1.5
python3 setup.py build --mpicc="mpicc -shared"
python3 setup.py install --user
- name: Run MPI4py
run: |
cd mpi4py-3.1.5/test
mpirun --map-by :OVERSUBSCRIBE -np 3 python3 ./main.py -v -f

0 comments on commit 9be4d0c

Please sign in to comment.