Skip to content

Commit 99d9bc6

Browse files
committed
Added make run_tutorials and updated mpi_examples.sh
1 parent 8b1f1d1 commit 99d9bc6

File tree

2 files changed

+11
-6
lines changed

2 files changed

+11
-6
lines changed

Makefile

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ dev-install_conda:
3131
conda env create -f environment-dev.yml && conda activate pylops_mpi && pip install -e .
3232

3333
lint:
34-
flake8 pylops_mpi/ tests/ examples/
34+
flake8 pylops_mpi/ tests/ examples/ tutorials/
3535

3636
tests:
3737
mpiexec -n $(NUM_PROCESSES) pytest tests/ --with-mpi
@@ -48,4 +48,8 @@ servedoc:
4848

4949
# Run examples using mpi
5050
run_examples:
51-
cd examples && sh mpi_examples.sh $(NUM_PROCESSES) && cd ..
51+
sh mpi_examples.sh examples $(NUM_PROCESSES)
52+
53+
# Run tutorials using mpi
54+
run_tutorials:
55+
sh mpi_examples.sh tutorials $(NUM_PROCESSES)

mpi_examples.sh

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,17 @@
11
#!/bin/bash
22

33
# Folder path containing example files
4+
cd $1 || exit;
45
folder_path="."
56

67
# Number of mpi processes
7-
num_processes=$1
8+
num_processes=$2
89

9-
if [ "$num_processes" -lt 2 ]; then
10-
echo "Number of processes should be greater than or equal to 2, $num_processes < 2"
11-
else
10+
if [ "$num_processes" -ge 2 ]; then
1211
for file in "$folder_path"/*.py; do
1312
echo "Running $file with $num_processes processes"
1413
mpiexec -n "$num_processes" python "$file"
1514
done
15+
else
16+
echo "Number of processes should be greater than or equal to 2, $num_processes < 2"
1617
fi

0 commit comments

Comments
 (0)