plan partitioning for mpi