Remove MPI from GPU runs