Add partial aggregation for sort-based shuffle read