In sorting, make number of partitions per node depend on the number of
nodes.