Check that ptychography reconstruction are working properly with multiple MPI ranks.
Approach
Ran the ptychography tests on MONA with 2 MPI ranks and 4 devices per rank. Uncovered a bug in the MPI code paths when multiple ranks run on the same node.
MPI tests now automatically divide devices on a single node amongst the the ranks instead of reusing the same devices.
Pre-Merge Checklists
Submitter
[x] Write a helpfully descriptive pull request title.
[x] Organize changes into logically grouped commits with descriptive commit messages.
[x] Document all new functions.
[ ] Click 'details' on the readthedocs check to view the updated docs.
[x] Write tests for new functions or explain why they are not needed.
[ ] Address any complaints from pep8speaks.
Reviewer
[ ] Actually read all of the code.
[ ] Run the new code yourself; the included tests should make this easy.
[ ] Write a summary of the changes as you understand them.
Purpose
Check that ptychography reconstruction are working properly with multiple MPI ranks.
Approach
Ran the ptychography tests on MONA with 2 MPI ranks and 4 devices per rank. Uncovered a bug in the MPI code paths when multiple ranks run on the same node.
MPI tests now automatically divide devices on a single node amongst the the ranks instead of reusing the same devices.
Pre-Merge Checklists
Submitter
Reviewer