nir: fix assign_io_var_locations for vertex inputs