ac/nir: Fix store_scratch with a non-full writemask