nir: Support vec8/vec16 in nir_lower_bit_size