- Merge _attn and _upcast_and_reordered_attn.