Add `ranks_per_bundle` parameter to extract_frame

desihub / gpu_specter

Scratch work for porting spectroperfectionism extractions to GPUs

BSD 3-Clause "New" or "Revised" License

2 stars 3 forks source link

Add `ranks_per_bundle` parameter to extract_frame #63

Closed dmargala closed 3 years ago

dmargala commented 3 years ago

This PR parametrizes the ranks_per_bundle variable that is used in extract_frame to decompose the mpi communicator into subgroups for extraction by bundles. For CPU extractions, this enables using multiple extraction groups instead of the current default of a single extraction group.