Might be good to add a timer for the partition_mesh step too, since that processing is potentially expensive. For the multi-volume case (i.e. when tag_to_elements is present), you could lump this code into it.
Might be good to add a timer for the partition_mesh step too, since that processing is potentially expensive. For the multi-volume case (i.e. when tag_to_elements is present), you could lump this code into it.
Questions for the review: