Respect parallel partitioning when 'collapsing' a sub-function

It seems that when collapsing a sub-function in parallel, the collapsed function does not respect the parallel partitioning of the parent. It would be helpful, and likely more efficient, if it respected the parent partitioning.

Both parent and collapsed function share the same partitioned mesh. The issue is ownership on the process boundary.