assign gpus to mpi processes