safeguard against num_rows=3 for one rank, for now.