Slightly generalize the code that handles shuffles of consecutive loads