Skip to content

Mpi support#42

Open
JamesMcClung wants to merge 4 commits into
mainfrom
mpi
Open

Mpi support#42
JamesMcClung wants to merge 4 commits into
mainfrom
mpi

Conversation

@JamesMcClung

Copy link
Copy Markdown
Owner

Adds an "mpi" option that sets up the runner to use mpi. Not sure if it actually works yet.

JamesMcClung and others added 4 commits May 29, 2026 08:32
dask-mpi + mpi4py, optional so pip install -e . stays MPI-free.

Co-Authored-By: Claude <noreply@anthropic.com>
PSC_PLOT_DASK_SCHEDULER=mpi calls dask_mpi.initialize(nthreads=1) so
mpirun -np N pins rank 0 -> scheduler, rank 1 -> client, rest -> workers.
Only rank 1 returns from initialize() and runs the rest of main(),
so no rank-gating logic is needed elsewhere.

Co-Authored-By: Claude <noreply@anthropic.com>
Co-Authored-By: Claude <noreply@anthropic.com>
@JamesMcClung JamesMcClung added the enhancement New feature or request label May 29, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant