st/mesa: assign driver locations for VS inputs for NIR before caching