Benchmark: CLI option to set subcomm size

desihub / gpu_specter

Scratch work for porting spectroperfectionism extractions to GPUs

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

2 stars 3 forks source link

Benchmark: CLI option to set subcomm size #73

Closed dmargala closed 2 years ago

dmargala commented 2 years ago

Add flexibility to the 30 frame benchmark by letting the user specify the number of MPI ranks per extraction group.

This is useful for testing the CPU version of gpu_specter on nodes with many CPU cores where it may be beneficial to split the MPI communicator into smaller extraction groups.

marcelo-alvarez commented 2 years ago

This looks good to me, the new flag functions in a similar way to that for desi_proc, and makes it easier to find the optimal subcomm size for a given system.