st/nir: Rework fixup_varying_slots