nir: fix lower_non_uniform_access pass