only use 2 MPI processes when testing, since GROMACS required one GPU per MPI process