gallivm/nir: add sample_mask_in support