add support for row lengths in PEs
needs more work, since we still assume NZ distribution is homogeneous between rows while distributing the NZs