The key improvement here IIUC is the reordering of the chunks themselves not just the new partitions(which un-swapped can make the diagram strictly inferior). Seems natural to keep it in the linearization step, as a improvement over potentially any non-optimal linearization step.
2 Likes