Instead of using `HashPartitioner`, we should use the same partitioner we used for creating in/out link blocks. This helps determine shuffle dependency. See my previous comment on subclassing partitioner.